| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx.py | 408 def is_leaf_module(self, m, qualname): member in TestFX.test_leaf_module.NoLeafModulesTracer
556 def is_leaf_module(self, module, name): member in TestFX.test_proxy_deepcopy_with_tracer.TestTracer
1119 def is_leaf_module(self, module, name): member in TestFX.test_deepcopy_graph_with_tracer_cls.TestTracer
1634 def is_leaf_module(self, m, module_qualified_name): member in TestFX.test_fx_create_arg.CreateArgTracer
1754 def is_leaf_module(self, m, module_qualified_name): member in TestFX.test_shape_prop_aggregate.RTTracer
2199 def is_leaf_module(self, m : torch.nn.Module, qualname : str): member in TestFX.test_replace_uses.LowerReluTracer
2528 def is_leaf_module(self, m: torch.nn.Module, module_qualified_name : str) -> bool: member in TestFX.test_trace_dict_int_keys.MyTracer
2547 def is_leaf_module(self, m: torch.nn.Module, module_qualified_name : str) -> bool: member in TestFX.test_trace_dict_proxy_keys.MyTracer
3279 def is_leaf_module(self, m: torch.nn.Module, module_qualified_name : str) -> bool: member in TestFX.test_delete_unused_submodules_leaf.MyCustomTracer
3352 def is_leaf_module(self, module, name): member in TestFX.test_tracing_graphmodules_as_leaf_submodules.LeafTracer
[all …]
|
| H A D | test_fx_experimental.py | 929 def is_leaf_module( member in TestFXExperimental.test_normalize_args.FunctionalTracer
1110 def is_leaf_module( member in TestFXExperimental.test_annotate_returns_with_schema.FunctionalTracer
|
| /aosp_15_r20/external/pytorch/test/package/package_a/ |
| H A D | test_all_leaf_modules_tracer.py | 7 def is_leaf_module(self, m, qualname): member in TestAllLeafModulesTracer
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | tracer.py | 36 def is_leaf_module(self, m: torch.nn.Module, module_qualified_name: str) -> bool: member in QuantizationTracer
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | fx_symbolic_graph_extractor.py | 39 def is_leaf_module( member in ModuleExpansionTracer
|
| /aosp_15_r20/external/pytorch/test/fx/ |
| H A D | test_dce_pass.py | 46 def is_leaf_module(self, m, qualname): member in TestDCE._run_dce_and_test.TestTracer
|
| /aosp_15_r20/external/pytorch/torch/fx/ |
| H A D | interpreter.py | 448 def is_leaf_module(self, _, __) -> bool: member in Transformer.__init__.TransformerTracer
|
| H A D | graph_module.py | 186 def is_leaf_module(self, _: torch.nn.Module, __: str) -> bool: member in _deserialize_graph_module.KeepModules
|
| H A D | _symbolic_trace.py | 428 def is_leaf_module(self, m: torch.nn.Module, module_qualified_name: str) -> bool: member in Tracer
|
| /aosp_15_r20/external/pytorch/torch/ao/ns/ |
| H A D | _numeric_suite_fx.py | 295 def is_leaf_module(self, m: torch.nn.Module, module_qualified_name: str) -> bool: member in NSTracer
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | proxy_tensor.py | 1711 def is_leaf_module(self, m: Module, module_qualified_name: str) -> bool: member in _ModuleStackTracer
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 598 def is_leaf_module(module): function
|