| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| group_by_window_test.py | 268 def reduce_func(_, xs): function in function:GroupByWindowTest.testReduceFuncError
278 grouping.group_by_window(lambda x, _: x % 2, reduce_func, 32))
286 def reduce_func(key, window): function in function:GroupByWindowTest.testConsumeWindowDatasetMoreThanOnce
301 reduce_func, 4))
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| grouping.py | 43 `reduce_func` is used to update the state every time an element is mapped to
52 the `init_func`, `reduce_func`, and `finalize_func` functions.
68 reduce_func,
75 `reduce_func` to at most `window_size_func(key)` elements matching the same
86 reduce_func: A function mapping a key and a dataset of up to `window_size`
90 batch, which will be passed to `reduce_func`. Mutually exclusive with
95 `reduce_func`. Mutually exclusive with `window_size`.
120 return _GroupByWindowDataset(dataset, key_func, reduce_func,
254 self._make_reduce_func(reducer.reduce_func, input_dataset)
264 reduce_func=self._reduce_func.function
463 def reduce_func(self): member in class:Reducer
[all...] |
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| group_by_reducer_dataset_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr("reduce_func", &reduce_func_));
114 TF_RETURN_IF_ERROR(b->AddFunction(ctx, reduce_func().name()));
147 AttrValue reduce_func; variable
148 b->BuildAttrValue(this->reduce_func(), &reduce_func);
173 {"reduce_func", reduce_func},
400 const NameAttrList& reduce_func() const { function in class:tensorflow::data::__anon45272::GroupByReducerDatasetOp::Dataset
|
| group_by_window_dataset_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr("reduce_func", &reduce_func_));
68 const NameAttrList& key_func, const NameAttrList& reduce_func,
78 reduce_func_(reduce_func),
138 AttrValue reduce_func; variable
139 b->BuildAttrValue(reduce_func_, &reduce_func);
159 {"reduce_func", reduce_func},
464 "`reduce_func` must return a single scalar of dtype "
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reduce_window_test.cc | 326 const auto reduce_func = [](float arg1, float arg2) { local
331 ReferenceUtil::ReduceWindow4DGeneric(input_array, 0.0f, reduce_func,
649 auto reduce_func = reducer == kAdd local
656 /*reduce_func=*/reduce_func,
1188 auto reduce_func = param.reducer == kAdd local
1381 auto reduce_func = param.reducer == kAdd local
[all...] |
| /external/python/cpython3/Modules/ |
| _pickle.c | 3940 PyObject *reduce_func = NULL; local
[all...] |
