Searched refs:num_input_comps (Results 1 – 1 of 1) sorted by relevance
| /honee/problems/torch/ |
| H A D | sgs_model_torch.cpp | 67 const PetscInt num_input_comps = 6, num_output_comps = 6; in ModelInference_Torch() local
81 num_nodes = input_size / num_input_comps; in ModelInference_Torch()
88 …at::from_blob((void *)dd_inputs_ptr, {num_nodes, num_input_comps}, {num_input_comps, 1}, nullptr, … in ModelInference_Torch()
91 …CXX(input_tensor = torch::from_blob((void *)dd_inputs_ptr, {num_nodes, num_input_comps}, options)); in ModelInference_Torch()
|