Searched refs:num_output_comps (Results 1 – 1 of 1) sorted by relevance
67 const PetscInt num_input_comps = 6, num_output_comps = 6; in ModelInference_Torch() local147 num_nodes = output_size / num_output_comps; in ModelInference_Torch()150 …Outputs_tensor = torch::from_blob((void *)dd_outputs_ptr, {num_nodes, num_output_comps}, options)); in ModelInference_Torch()