Home
last modified time | relevance | path

Searched defs:check_fn (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dcache_entry.h41 py::object check_fn; member
H A Dextra_state.cpp166 py::handle check_fn = py::handle(guarded_code).attr("check_fn"); in create_cache_entry() local
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_data_parallel.py816 def __init__(self, data, check_fn): argument
830 def check_fn(self_): function
H A Dtest_dynamo_distributed.py574 check_fn = lambda submodule: isinstance( # noqa: E731 function
744 check_fn = lambda submodule: isinstance( # noqa: E731 function
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_profiler.py116 def check_fn(): function
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Ddist_util.py121 def check_fn(submodule): function
/aosp_15_r20/external/pytorch/torch/jit/
H A Dannotations.py181 def check_fn(fn, loc): function
/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_checkpoint/
H A Dcheckpoint_wrapper.py252 check_fn=lambda _: True, argument
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_checkpoint_wrapper.py247 def check_fn(l): function
H A Dtest_fsdp_meta.py132 def check_fn(k): function
/aosp_15_r20/external/toolchain-utils/toolchain_utils_githooks/
H A Dcheck-presubmit.py1047 def run_check(check_fn): argument
H A Dcheck-presubmit1047 def run_check(check_fn): argument
/aosp_15_r20/external/linux-kselftest/tools/testing/selftests/arm64/abi/
H A Dsyscall-abi.c316 typedef int (*check_fn)(struct syscall_cfg *cfg, int sve_vl, int sme_vl, typedef
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dtrt_convert_test.py391 def _CheckTrtOps(self, concrete_func, check_fn=None, num_engines=1): argument
/aosp_15_r20/external/libpng/contrib/tools/
H A Dgenpng.c179 shape_fn_ptr check_fn; member
/aosp_15_r20/packages/providers/MediaProvider/jni/
Dnode_test.cpp486 auto check_fn = [&](const node* lhs_node, const node* rhs_node) { in TEST_F() local
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dops.py110 def check_fn(x, is_optional=is_optional): function
/aosp_15_r20/system/apex/apexd/
H A Dapexd.cpp727 auto check_fn = [&](const std::vector<std::string>& mount_points) in VerifyPackagesStagedInstall() local
3369 auto check_fn = in VerifyPackageNonStagedInstall() local
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dpattern_matcher.py1231 def check_fn(match: Match) -> bool: function
/aosp_15_r20/external/dtc/
H A Dchecks.c29 typedef void (*check_fn)(struct check *c, struct dt_info *dti, struct node *node); typedef
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe_src.cc231 #define PARSE_VALUE(fn_name, type, check_fn, parse_fn) \ argument