Searched refs:input_tensor (Results 1 – 1 of 1) sorted by relevance
66 torch::Tensor input_tensor, output_tensor; in ModelInference_Torch() local87 PetscCallCXX(input_tensor = in ModelInference_Torch()91 …PetscCallCXX(input_tensor = torch::from_blob((void *)dd_inputs_ptr, {num_nodes, num_input_comps}, … in ModelInference_Torch()97 … PetscCallCXX(input_tensor_ptr = (double *)input_tensor.contiguous().to(torch::kCPU).data_ptr()); in ModelInference_Torch()110 PetscCallCXX(output_tensor = model.forward({input_tensor}).toTensor()); in ModelInference_Torch()