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)
|