| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | grouping.py | 61 reduce_func, argument
303 def _make_reduce_func(self, reduce_func, input_dataset): argument
413 def __init__(self, init_func, reduce_func, finalize_func): argument
423 def reduce_func(self): member in Reducer
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | decompose_reduce_dataset.cc | 141 OpBuilder builder, ReduceDatasetOp reduce_dataset, FuncOp reduce_func, in CreateOptionalDatasetIf()
216 FuncOp reduce_func, WhileRegionOp dataset_while, in PopulateDatasetWhileBody()
278 func::FuncOp reduce_func = in DecomposeReduceDatasetInFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | group_by_window_test.py | 284 def reduce_func(_, xs): function
304 def reduce_func(key, window): function
|
| H A D | checkpoint_test.py | 489 reduce_func = lambda _, x: True function
510 reduce_func = lambda _, x: x function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | reference_util.cc | 147 const std::function<float(float, float)>& reduce_func, in ReduceWindow1DGeneric() argument
235 const std::function<float(float, float)>& reduce_func, in ReduceWindow4DGeneric() argument
248 const std::function<float(float, float)>& reduce_func, in ReduceWindow4DGeneric() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/types/ |
| H A D | distribute.py | 29 def reduce(self, initial_state, reduce_func): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | tf_record_writer_test.py | 124 def reduce_func(key, dataset): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reduce_window_test.cc | 347 const auto reduce_func = [](float arg1, float arg2) { in XLA_TEST_P() local
663 auto reduce_func = reducer == kAdd in DoIt() local
1548 auto reduce_func = param.reducer == kAdd in XLA_TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | group_by_reducer_dataset_op.cc | 159 AttrValue reduce_func; in AsGraphDefInternal() local
|
| H A D | group_by_window_dataset_op.cc | 158 AttrValue reduce_func; in AsGraphDefInternal() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | dataset_ops.py | 2572 def reduce(self, initial_state, reduce_func, name=None): argument
2902 reduce_func, argument
5841 reduce_func, argument
5892 def _make_reduce_func(self, reduce_func, input_dataset): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | reduce.cc | 614 std::function<T(T, T)> reduce_func; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | input_lib.py | 433 def reduce(self, initial_state, reduce_func): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Indexing.cu | 1007 const func_t& reduce_func, in index_reduce_func_cuda_impl()
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_control_flow.py | 241 def reduce_func(*operands): function
|
| /aosp_15_r20/external/python/cpython3/Modules/ |
| D | _pickle.c | 4274 PyObject *reduce_func = NULL; in save() local
|