Home
last modified time | relevance | path

Searched refs:reduce_func (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dgrouping.py69 reduce_func, argument
121 return _GroupByWindowDataset(dataset, key_func, reduce_func,
280 self._make_reduce_func(reducer.reduce_func, input_dataset)
290 reduce_func=self._reduce_func.function,
313 def _make_reduce_func(self, reduce_func, input_dataset): argument
326 reduce_func,
398 def __init__(self, input_dataset, key_func, reduce_func, window_size_func): argument
402 self._make_reduce_func(reduce_func, input_dataset)
410 reduce_func=self._reduce_func.function,
441 def _make_reduce_func(self, reduce_func, input_dataset): argument
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py43 reduce_func=lambda x, y: x + y,
62 reduce_func=reduce_fn,
78 reduce_func=lambda x, y: x + y[0],
100 reduce_func=lambda x, y: _sparse(x.values[0] + y.values[0]),
122 reduce_func=reduce_fn,
142 reduce_func=lambda x, y: constant_op.constant(1, dtype=dtypes.int64),
157 reduce_func=lambda x, y: x + y,
171 reduce_func=lambda x, y: x + y,
Dgroup_by_window_test.py277 def reduce_func(_, xs): function
287 grouping.group_by_window(lambda x, _: x % 2, reduce_func, 32))
296 def reduce_func(key, window): function
311 reduce_func, 4))
Dtf_record_writer_test.py128 def reduce_func(key, dataset): function
137 grouping.group_by_window(lambda i, _: i % 2, reduce_func,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-reducer.pbtxt14 name: "reduce_func"
19 …argspec: "args=[\'self\', \'init_func\', \'reduce_func\', \'finalize_func\'], varargs=None, keywor…
Dtensorflow.distribute.-distributed-dataset.pbtxt14 …argspec: "args=[\'self\', \'initial_state\', \'reduce_func\'], varargs=None, keywords=None, defaul…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-reducer.pbtxt14 name: "reduce_func"
19 …argspec: "args=[\'self\', \'init_func\', \'reduce_func\', \'finalize_func\'], varargs=None, keywor…
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DExperimentalGroupByWindowDataset.pbtxt28 name: "reduce_func"
91 name: "reduce_func"
DGroupByWindowDataset.pbtxt28 name: "reduce_func"
DGroupByReducerDataset.pbtxt36 name: "reduce_func"
DExperimentalGroupByReducerDataset.pbtxt36 name: "reduce_func"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DExperimentalGroupByWindowDataset.pbtxt28 name: "reduce_func"
91 name: "reduce_func"
DGroupByWindowDataset.pbtxt28 name: "reduce_func"
DGroupByReducerDataset.pbtxt36 name: "reduce_func"
DExperimentalGroupByReducerDataset.pbtxt36 name: "reduce_func"
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py469 reduce_func = lambda _, x: True function
474 (stateful_key_func, init_func, reduce_func, finalize_func),
475 (key_func, stateful_init_func, reduce_func, finalize_func),
477 (key_func, init_func, reduce_func, stateful_finalize_func),
479 for key_func, init_func, reduce_func, finalize_func in test_cases:
481 reducer = grouping.Reducer(init_func, reduce_func, finalize_func)
490 reduce_func = lambda _, x: x function
495 (stateful_key_func, reduce_func, window_func),
497 (key_func, reduce_func, stateful_window_func),
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExperimentalGroupByReducerDataset.pbtxt42 building a closure for `reduce_func`.
46 name: "reduce_func"
Dapi_def_GroupByReducerDataset.pbtxt42 building a closure for `reduce_func`.
46 name: "reduce_func"
/external/tensorflow/tensorflow/python/types/
Ddistribute.py32 def reduce(self, initial_state, reduce_func): argument
/external/llvm-project/openmp/runtime/test/misc_bugs/
Dteams-reduction.c46 size_t reduce_size, void *reduce_data, void *reduce_func,
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dgroup_by_reducer_serialization_test.py38 reduce_func=lambda x, y: x + y,
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc146 const std::function<float(float, float)>& reduce_func, in ReduceWindow1DGeneric() argument
167 val = reduce_func(val, operand[i0_base + i0_win]); in ReduceWindow1DGeneric()
234 const std::function<float(float, float)>& reduce_func, in ReduceWindow4DGeneric() argument
240 operand, init, reduce_func, window, stride, in ReduceWindow4DGeneric()
247 const std::function<float(float, float)>& reduce_func, in ReduceWindow4DGeneric() argument
284 val = reduce_func( in ReduceWindow4DGeneric()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc159 AttrValue reduce_func; in AsGraphDefInternal() local
160 b->BuildAttrValue(captured_reduce_func_->func(), &reduce_func); in AsGraphDefInternal()
185 {"reduce_func", reduce_func}, in AsGraphDefInternal()
Dgroup_by_window_dataset_op.cc158 AttrValue reduce_func; in AsGraphDefInternal() local
159 b->BuildAttrValue(captured_reduce_func_->func(), &reduce_func); in AsGraphDefInternal()
179 {"reduce_func", reduce_func}, in AsGraphDefInternal()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc349 const auto reduce_func = [](float arg1, float arg2) { in XLA_TEST_P() local
354 ReferenceUtil::ReduceWindow4DGeneric(input_array, 0.0f, reduce_func, in XLA_TEST_P()
665 auto reduce_func = reducer == kAdd in DoIt() local
672 /*reduce_func=*/reduce_func, in DoIt()
1475 auto reduce_func = param.reducer == kAdd in XLA_TEST_P() local
1481 /*reduce_func=*/reduce_func, in XLA_TEST_P()

123