Searched refs:torch_to_onnx_output (Results 1 – 1 of 1) sorted by relevance
2054 std::unordered_map<std::string, std::string> torch_to_onnx_output; in ONNXShapeTypeInference() local2084 torch_to_onnx_output[n->output(i)->debugName()] = in ONNXShapeTypeInference()2174 inferred_shape_data.find(torch_to_onnx_output[output->debugName()]); in ONNXShapeTypeInference()