Home
last modified time | relevance | path

Searched defs:reduce_fn (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dreduce_test.py50 def reduce_fn(state, value): function
63 def reduce_fn(state, value): function
80 def reduce_fn(state, _): function
93 def reduce_fn(_, value): function
110 def reduce_fn(state, value): function
143 def reduce_fn(state, value): function
162 def reduce_fn(state, value): function
227 def reduce_fn(state, value): function
Doptional_test.py67 def reduce_fn(x, y): function
108 def reduce_fn(x, y): function
Dmap_test.py119 def reduce_fn(x, y): function
/external/tensorflow/tensorflow/python/data/util/
Dsparse_test.py51 def reduce_fn(x, y): function
106 def reduce_fn(x, y): function
148 def reduce_fn(x, y): function
194 def reduce_fn(x, y): function
222 def reduce_fn(x, y): function
247 def reduce_fn(x, y): function
Dstructure_test.py85 def reduce_fn(x, y): function
175 def reduce_fn(x, y): function
222 def reduce_fn(x, y): function
245 def reduce_fn(x, y): function
279 def reduce_fn(x, y): function
329 def reduce_fn(x, y): function
382 def reduce_fn(x, y): function
420 def reduce_fn(x, y): function
457 def reduce_fn(x, y): function
486 def reduce_fn(x, y): function
Drandom_seed_test.py94 def reduce_fn(x, y): function
Dstructure.py340 def reduce_fn(state, value): function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py53 def reduce_fn(x, y): function
110 def reduce_fn(x, _): function
178 def reduce_fn(state, value): function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Doptimization_test.py68 def reduce_fn(key, bucket): function
92 def reduce_fn(x, y): function
Dnoop_elimination_test.py77 def reduce_fn(result, case): function
Dmap_fusion_test.py55 def reduce_fn(x, y): function
Dfilter_fusion_test.py50 def reduce_fn(x, y): function
Dmap_parallelization_test.py46 def reduce_fn(x, y): function
Dmap_and_filter_fusion_test.py65 def reduce_fn(x, y): function
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops_test.py239 def reduce_fn(): function
899 def reduce_fn(): function
1017 def reduce_fn(v): function
Dinput_lib_test.py1179 def sum_while_loop(iterator, reduce_fn): argument
Dinput_lib.py931 def reduce(self, initial_state, reduce_fn): argument
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/
Ddynamic_sharding_test.py97 def reduce_fn(_, window): function
/external/tensorflow/tensorflow/python/eager/
Dfunction_gradients_test.py938 def reduce_fn(x): function
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_test.py1548 def reduce_fn(t): function
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py629 def reduce_fn(input_tensor, axis, keepdims): function
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc338 XlaComputation reduce_fn = b->BuildAndNoteError(); in XLA_TEST_P() local
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py235 def reduce_fn(p, q): function