Home
last modified time | relevance | path

Searched defs:key_func (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dgrouping.py29 def group_by_reducer(key_func, reducer): argument
60 def group_by_window(key_func, argument
263 def __init__(self, input_dataset, key_func, reducer): argument
283 def _make_key_func(self, key_func, input_dataset): argument
/external/pytorch/tools/jit/
Dgen_unboxing.py165 def key_func(fn: NativeFunction | NativeFunctionsGroup) -> str: function
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py485 key_func = lambda _: math_ops.cast(0, dtypes.int64) function
508 key_func = lambda _: math_ops.cast(0, dtypes.int64) function
/external/mesa3d/src/panfrost/compiler/
Dbifrost_isa.py410 key_func = lambda x: x[1:] function
/external/pytorch/torchgen/
Dgen_executorch.py297 def key_func( function
Dgen.py2487 def key_func( function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc155 AttrValue key_func; in AsGraphDefInternal() local
Dgroup_by_window_dataset_op.cc156 AttrValue key_func; in AsGraphDefInternal() local
/external/pytorch/tools/autograd/
Dgen_python_functions.py524 def key_func( function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/
Ddata_service_ops_test.py1228 def key_func(x): function
/external/angle/build/
Dlocale_tool.py148 def _SortListSubRange(input_list, start, end, key_func): argument
/external/cronet/stable/build/
Dlocale_tool.py148 def _SortListSubRange(input_list, start, end, key_func): argument
/external/cronet/tot/build/
Dlocale_tool.py148 def _SortListSubRange(input_list, start, end, key_func): argument
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py2901 key_func, argument
5840 key_func, argument
5878 def _make_key_func(self, key_func, input_dataset): argument