Searched refs:parse_tensor_constants (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
H A D | irparser.cpp | 29 bool parse_tensor_constants); 34 bool parse_tensor_constants) in IRParser() argument 42 parse_tensor_constants_(parse_tensor_constants) {} in IRParser() 112 bool parse_tensor_constants) { in parseIR() argument 113 torch::jit::IRParser p(str, graph, vmap, parse_tensor_constants); in parseIR() 120 bool parse_tensor_constants) { in parseIR() argument 122 parseIR(str, graph, vmap, parse_tensor_constants); in parseIR()
|
H A D | irparser.h | 21 bool parse_tensor_constants = false); 35 bool parse_tensor_constants = false);
|
/aosp_15_r20/external/pytorch/test/onnx/ |
H A D | test_pytorch_jit_onnx.py | 56 def run_test(self, graph_ir, example_inputs, parse_tensor_constants=False): argument 57 graph = torch._C.parse_ir(graph_ir, parse_tensor_constants) 104 self.run_test(graph_ir, (a, b), parse_tensor_constants=True)
|
/aosp_15_r20/external/pytorch/torch/utils/jit/ |
H A D | log_extract.py | 43 graph = torch._C.parse_ir(ir, parse_tensor_constants=True)
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
H A D | init.cpp | 1808 [](const std::string& input, bool parse_tensor_constants) { in initJITBindings() argument 1810 parseIR(input, &*graph, parse_tensor_constants); in initJITBindings()
|
/aosp_15_r20/external/pytorch/torch/_C/ |
H A D | __init__.pyi.in | 522 def parse_ir(input: str, parse_tensor_constants: _bool = False) -> Graph: ...
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_jit.py | 8226 g_parsed = parse_ir(g, parse_tensor_constants=True) 8238 parse_ir(g, parse_tensor_constants=False)
|