Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dreduce_test.py54 def reduce_fn(state, value): function
67 def reduce_fn(state, value): function
84 def reduce_fn(state, _): function
97 def reduce_fn(_, value): function
114 def reduce_fn(state, value): function
147 def reduce_fn(state, value): function
166 def reduce_fn(state, value): function
231 def reduce_fn(state, value): function
Doptional_test.py71 def reduce_fn(x, y): function
112 def reduce_fn(x, y): function
Dmap_test.py121 def reduce_fn(x, y): function
/external/tensorflow/tensorflow/python/data/util/
Dsparse_test.py55 def reduce_fn(x, y): function
110 def reduce_fn(x, y): function
152 def reduce_fn(x, y): function
198 def reduce_fn(x, y): function
226 def reduce_fn(x, y): function
251 def reduce_fn(x, y): function
Dstructure_test.py90 def reduce_fn(x, y): function
176 def reduce_fn(x, y): function
223 def reduce_fn(x, y): function
246 def reduce_fn(x, y): function
280 def reduce_fn(x, y): function
330 def reduce_fn(x, y): function
383 def reduce_fn(x, y): function
421 def reduce_fn(x, y): function
458 def reduce_fn(x, y): function
487 def reduce_fn(x, y): function
Drandom_seed_test.py98 def reduce_fn(x, y): function
Dstructure.py335 def reduce_fn(state, value): function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dgroup_by_reducer_test.py57 def reduce_fn(x, y): function
114 def reduce_fn(x, _): function
186 def reduce_fn(state, value): function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Doptimization_test.py73 def reduce_fn(key, bucket): function
97 def reduce_fn(x, y): function
Dnoop_elimination_test.py81 def reduce_fn(result, case): function
Dmap_fusion_test.py59 def reduce_fn(x, y): function
Dfilter_fusion_test.py54 def reduce_fn(x, y): function
Dmap_parallelization_test.py50 def reduce_fn(x, y): function
Dmap_and_filter_fusion_test.py69 def reduce_fn(x, y): function
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/benchmarks/
Dnormalization_adapt_benchmark.py35 def reduce_fn(state, values): function
Ddiscretization_adapt_benchmark.py36 def reduce_fn(state, values, epsilon=EPSILON): function
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops_test.py241 def reduce_fn(): function
852 def reduce_fn(): function
948 def reduce_fn(v): function
Dinput_lib_test.py1139 def sum_while_loop(iterator, reduce_fn): argument
Dinput_lib.py1053 def reduce(self, initial_state, reduce_fn): argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_test.py1139 def reduce_fn(t): function
/external/tensorflow/tensorflow/python/eager/
Dfunction_gradients_test.py941 def reduce_fn(x): function
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py638 def reduce_fn(input_tensor, axis, keepdims): function
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_window_test.cc340 XlaComputation reduce_fn = b->BuildAndNoteError(); in XLA_TEST_P() local
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py239 def reduce_fn(p, q): function