Searched defs:extra_check (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/ |
| H A D | _common.py | 12 def _sharded_op_common(op, early_stop_func, extra_check): argument
63 op, early_stop_func=None, extra_check=None, customized_func=None argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | freezing_patterns.py | 99 def register_freezing_graph_pattern(pattern, extra_check=_return_true, pass_number=0): argument
107 def register_binary_folding_pattern(pattern, extra_check=_return_true): argument
|
| H A D | reinplace.py | 33 extra_check: Callable[[torch.fx.Node], bool] = lambda node: True variable in InplaceableOp
|
| H A D | post_grad.py | 192 def register_lowering_pattern(pattern, extra_check=_return_true, pass_number=1): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_micro_gemm.py | 185 extra_check: Optional[Callable[..., bool]] = None variable in CppMicroGemmConfig
210 extra_check=None, argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharding_spec/chunk_sharding_spec_ops/ |
| H A D | _common.py | 32 op, early_stop_func=None, extra_check=None, customized_func=None argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_custom_post_grad_passes.py | 94 pattern, extra_check, custom_pass_dict argument
|
| H A D | test_pattern_matcher.py | 1002 def extra_check(match): function
|
| /aosp_15_r20/external/executorch/exir/tests/ |
| H A D | test_memory_planning.py | 142 def extra_check( member in ModuleListArg
|
| /aosp_15_r20/external/rust/android-crates-io/crates/rusqlite/ |
| D | Cargo.toml | 171 extra_check = [] qkey
|
| /aosp_15_r20/art/libdexfile/dex/ |
| H A D | dex_file_verifier.cc | 477 ExtraCheckFn extra_check) { in VerifyTypeDescriptor()
|