Searched refs:output_tensor_ptr (Results 1 – 1 of 1) sorted by relevance
129 double *output_tensor_ptr; in ModelInference_Torch() local133 … PetscCallCXX(output_tensor_ptr = (double *)output_tensor.contiguous().to(torch::kCPU).data_ptr()); in ModelInference_Torch()137 printf("%f\n", output_tensor_ptr[i]); in ModelInference_Torch()140 PetscCall(PetscArraycpy(dd_outputs_ptr, output_tensor_ptr, output_size)); in ModelInference_Torch()