Searched defs:reduce_func (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
H A Dbucketing_test.py108 def reduce_func(_, xs): function in function:GroupByWindowTest.testReduceFuncError
119 grouping.group_by_window(lambda x, _: x % 2, reduce_func,
132 def reduce_func(key, window): function in function:GroupByWindowTest.testConsumeWindowDatasetMoreThanOnce
147 reduce_func, 4))
/external/tensorflow/tensorflow/core/kernels/data/
H A Dgroup_by_window_dataset_op.cc36 OP_REQUIRES_OK(ctx, ctx->GetAttr("reduce_func", &reduce_func_));
99 const NameAttrList& key_func, const NameAttrList& reduce_func,
109 reduce_func_(reduce_func),
165 AttrValue reduce_func; variable
166 b->BuildAttrValue(reduce_func_, &reduce_func);
186 {"reduce_func", reduce_func},
471 "`reduce_func` must return a single scalar of dtype "
98 Dataset(OpKernelContext* ctx, const DatasetBase* input, const NameAttrList& key_func, const NameAttrList& reduce_func, const NameAttrList& window_size_func, std::unique_ptr<CapturedFunction> captured_key_func, std::unique_ptr<CapturedFunction> captured_reduce_func, std::unique_ptr<CapturedFunction> captured_window_size_func, const DataTypeVector& output_types, const std::vector<PartialTensorShape>& output_shapes) argument
/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util.cc194 const std::function<float(float, float)>& reduce_func,
199 operand, init, reduce_func, window, stride,
206 const std::function<float(float, float)>& reduce_func,
228 val = reduce_func(val, operand[i0_base + i0_win]);
332 const std::function<float(float, float)>& reduce_func,
338 operand, init, reduce_func, window, stride,
345 const std::function<float(float, float)>& reduce_func,
383 val = reduce_func(
192 ReduceWindow1DGeneric( const tensorflow::gtl::ArraySlice<float>& operand, float init, const std::function<float(float, float)>& reduce_func, const tensorflow::gtl::ArraySlice<int64>& window, const tensorflow::gtl::ArraySlice<int64>& stride, Padding padding) argument
204 ReduceWindow1DGeneric( const tensorflow::gtl::ArraySlice<float>& operand, float init, const std::function<float(float, float)>& reduce_func, const tensorflow::gtl::ArraySlice<int64>& window, const tensorflow::gtl::ArraySlice<int64>& stride, const tensorflow::gtl::ArraySlice<std::pair<int64, int64>>& padding) argument
330 ReduceWindow4DGeneric( const Array4D<float>& operand, float init, const std::function<float(float, float)>& reduce_func, const tensorflow::gtl::ArraySlice<int64>& window, const tensorflow::gtl::ArraySlice<int64>& stride, Padding padding) argument
343 ReduceWindow4DGeneric( const Array4D<float>& operand, float init, const std::function<float(float, float)>& reduce_func, const tensorflow::gtl::ArraySlice<int64>& window, const tensorflow::gtl::ArraySlice<int64>& stride, const tensorflow::gtl::ArraySlice<std::pair<int64, int64>>& padding) argument
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreduce_window_test.cc280 const auto reduce_func = [](float arg1, float arg2) { local
285 ReferenceUtil::ReduceWindow4DGeneric(input_array, 0.0f, reduce_func,
606 auto reduce_func = param.reducer == kAdd local
613 /*reduce_func=*/reduce_func,
1242 auto reduce_func = param.reducer == kAdd local
1248 /*reduce_func=*/reduce_func,
/external/python/cpython3/Modules/
H A D_pickle.c3769 PyObject *reduce_func = NULL; local
3871 reduce_func = PyDict_GetItemWithError(st->dispatch_table,
3873 if (reduce_func == NULL) {
3881 Py_INCREF(reduce_func);
3884 reduce_func = PyObject_GetItem(self->dispatch_table,
3886 if (reduce_func == NULL) {
3893 if (reduce_func != NULL) {
3895 reduce_value = _Pickle_FastCall(reduce_func, obj);
3915 reduce_func = _PyObject_GetAttrId(obj, &PyId___reduce_ex__);
3916 if (reduce_func !
[all...]

Completed in 715 milliseconds