Searched refs:torch_to_onnx_input (Results 1 – 1 of 1) sorted by relevance
2053 std::unordered_map<std::string, std::string> torch_to_onnx_input; in ONNXShapeTypeInference() local2079 torch_to_onnx_input[n->input(i)->debugName()] = in ONNXShapeTypeInference()2094 torch_to_onnx_input.find(input->debugName()); in ONNXShapeTypeInference()2095 if (onnx_output_name != torch_to_onnx_input.end()) { in ONNXShapeTypeInference()