Searched refs:output_tensor (Results 1 – 1 of 1) sorted by relevance
66 torch::Tensor input_tensor, output_tensor; in ModelInference_Torch() local110 PetscCallCXX(output_tensor = model.forward({input_tensor}).toTensor()); in ModelInference_Torch()133 … PetscCallCXX(output_tensor_ptr = (double *)output_tensor.contiguous().to(torch::kCPU).data_ptr()); in ModelInference_Torch()151 PetscCallCXX(DD_Outputs_tensor.copy_(output_tensor)); in ModelInference_Torch()