Searched refs:DD_Outputs_tensor (Results 1 – 1 of 1) sorted by relevance
144 torch::Tensor DD_Outputs_tensor; in ModelInference_Torch() local150 …PetscCallCXX(DD_Outputs_tensor = torch::from_blob((void *)dd_outputs_ptr, {num_nodes, num_output_c… in ModelInference_Torch()151 PetscCallCXX(DD_Outputs_tensor.copy_(output_tensor)); in ModelInference_Torch()