| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | lazy.py | 28 local_metadata, argument
212 local_metadata, argument
|
| H A D | instancenorm.py | 62 local_metadata, argument
|
| H A D | batchnorm.py | 112 local_metadata, argument
|
| H A D | module.py | 2299 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | spectral_norm.py | 206 local_metadata, argument
253 def __call__(self, module, state_dict, prefix, local_metadata) -> None: argument
|
| H A D | parametrizations.py | 383 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | linear.py | 81 local_metadata, argument
235 local_metadata, argument
|
| H A D | embedding_ops.py | 71 local_metadata, argument
|
| H A D | functional_modules.py | 208 local_metadata, argument
|
| H A D | conv.py | 195 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| H A D | __init__.py | 412 def state_dict_hook(module, destination, prefix, local_metadata): argument
430 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/ |
| H A D | linear.py | 65 local_metadata, argument
173 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_module_hooks.py | 564 local_metadata, argument
578 local_metadata, argument
638 local_metadata, argument
656 local_metadata, argument
875 def linear_state_dict_post_hook(module, state_dict, prefix, local_metadata): argument
|
| H A D | test_load_state_dict.py | 196 local_metadata, argument
238 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| H A D | rnn.py | 73 local_metadata, argument
292 local_metadata, argument
1177 local_metadata, argument
|
| H A D | linear.py | 81 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/dynamic/ |
| H A D | linear.py | 79 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | graph_module.py | 170 local_metadata, argument
|
| H A D | _lower_to_native_backend.py | 428 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | observer.py | 276 local_metadata, argument
1340 local_metadata, argument
|
| H A D | fake_quantize.py | 277 local_metadata, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/ |
| H A D | utils.py | 165 local_metadata, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/ |
| H A D | single_machine.cc | 259 auto local_metadata = std::make_shared<RunMetadata>(); in RunWithTimeout() local
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | torchbench.py | 37 def state_dict_hook(module, destination, prefix, local_metadata): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/ |
| H A D | conv_fused.py | 307 local_metadata, argument
|