Home
last modified time | relevance | path

Searched defs:reduce_func (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dgrouping.py64 reduce_func, argument
304 def _make_reduce_func(self, reduce_func, input_dataset): argument
396 def __init__(self, init_func, reduce_func, finalize_func): argument
406 def reduce_func(self): member in Reducer
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dgroup_by_window_test.py288 def reduce_func(_, xs): function
308 def reduce_func(key, window): function
Dcheckpoint_test.py493 reduce_func = lambda _, x: True function
514 reduce_func = lambda _, x: x function
/external/tensorflow/tensorflow/python/types/
Ddistribute.py32 def reduce(self, initial_state, reduce_func): argument
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dtf_record_writer_test.py128 def reduce_func(key, dataset): function
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc146 const std::function<float(float, float)>& reduce_func, in ReduceWindow1DGeneric() argument
234 const std::function<float(float, float)>& reduce_func, in ReduceWindow4DGeneric() argument
247 const std::function<float(float, float)>& reduce_func, in ReduceWindow4DGeneric() argument
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc159 AttrValue reduce_func; in AsGraphDefInternal() local
Dgroup_by_window_dataset_op.cc158 AttrValue reduce_func; in AsGraphDefInternal() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc349 const auto reduce_func = [](float arg1, float arg2) { in XLA_TEST_P() local
665 auto reduce_func = reducer == kAdd in DoIt() local
1475 auto reduce_func = param.reducer == kAdd in XLA_TEST_P() local
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py2239 def reduce(self, initial_state, reduce_func): argument
2552 reduce_func, argument
5356 def __init__(self, input_dataset, key_func, reduce_func, window_size_func): argument
5401 def _make_reduce_func(self, reduce_func, input_dataset): argument
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py540 def reduce(self, initial_state, reduce_func): argument
/external/python/cpython3/Modules/
D_pickle.c4258 PyObject *reduce_func = NULL; in save() local