Home
last modified time | relevance | path

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

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