Home
last modified time | relevance | path

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

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