Home
last modified time | relevance | path

Searched refs:parse_tensor_constants (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/ir/
H A Dirparser.cpp29 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 Dirparser.h21 bool parse_tensor_constants = false);
35 bool parse_tensor_constants = false);
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_jit_onnx.py56 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 Dlog_extract.py43 graph = torch._C.parse_ir(ir, parse_tensor_constants=True)
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dinit.cpp1808 [](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.in522 def parse_ir(input: str, parse_tensor_constants: _bool = False) -> Graph: ...
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py8226 g_parsed = parse_ir(g, parse_tensor_constants=True)
8238 parse_ir(g, parse_tensor_constants=False)