Searched refs:method_graph (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/ao/quantization/ |
| D | quantize_jit.py | 96 method_graph = model._c._get_method(method_name).graph
97 torch._C._jit_pass_inline(method_graph)
|
| /external/executorch/docs/source/ |
| D | bundled-io.md | 132 method_graph = export(
139 et_program = to_edge(method_graph).to_executorch()
339 method_graph = export(
345 et_program = to_edge(method_graph).to_executorch()
475 method_graph = export(
481 et_program = to_edge(method_graph).to_executorch()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | vulkan_rewrite.cpp | 499 auto method_graph = method.graph(); in vulkanRunCanonicalOptimizations() local
500 runOptimization(method_graph, false /* no loop unrolling */); in vulkanRunCanonicalOptimizations()
|
| /external/pytorch/torch/onnx/ |
| D | utils.py | 924 def _get_param_count_list(method_graph, args_params): argument
926 for input_, arg_params_ in zip(method_graph.inputs(), args_params):
978 method_graph = module._get_method("forward").graph
980 param_count_list = _get_param_count_list(method_graph, args_params)
983 method_graph, tuple(in_vars), param_count_list, False, False
|
| /external/pytorch/torch/_export/ |
| D | converter.py | 32 def _get_param_count_list(method_graph, args_params): argument
34 for input_, arg_params_ in zip(method_graph.inputs(), args_params):
89 method_graph = module._get_method("forward").graph
91 param_count_list = _get_param_count_list(method_graph, args_params)
94 method_graph, tuple(in_vars), param_count_list, False, False
|