| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | reduce_test.py | 54 def reduce_fn(state, value): function
61 result = ds.reduce(constant_op.constant(0, dtype=dtypes.int64), reduce_fn)
67 def reduce_fn(state, value): function
75 reduce_fn)
84 def reduce_fn(state, _): function
89 result = ds.reduce(np.int64(0), reduce_fn)
97 def reduce_fn(_, value): function
108 result = ds.reduce(make_sparse_fn(0), reduce_fn)
114 def reduce_fn(state, value): function
131 result = ds.reduce(map_fn(0), reduce_fn)
[all …]
|
| D | optional_test.py | 71 def reduce_fn(x, y): function
77 return functools.reduce(reduce_fn, cases, [])
112 def reduce_fn(x, y): function
119 return functools.reduce(reduce_fn, cases, [])
|
| D | map_test.py | 113 def reduce_fn(x, y): function
118 return functools.reduce(reduce_fn, cases, [])
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | group_by_reducer_test.py | 56 def reduce_fn(x, y): function
62 reduce_func=reduce_fn,
113 def reduce_fn(x, _): function
122 reduce_func=reduce_fn,
185 def reduce_fn(state, value): function
193 reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
|
| D | optimize_dataset_test.py | 75 def reduce_fn(key, bucket): function
80 grouping.group_by_window(lambda _: 0, reduce_fn, 10))
99 def reduce_fn(x, y): function
104 return functools.reduce(reduce_fn, cases, [])
120 def reduce_fn(x, y): function
126 return functools.reduce(reduce_fn, cases, [])
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/benchmarks/ |
| D | discretization_adapt_benchmark.py | 39 def reduce_fn(state, values, epsilon=EPSILON): function
68 state = ds.reduce((np.zeros((1, 2)),), reduce_fn)
|
| D | normalization_adapt_benchmark.py | 38 def reduce_fn(state, values): function
79 k, n, ex, ex2 = ds.reduce((0.0, 0, 0.0, 0.0), reduce_fn)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | noop_elimination_test.py | 78 def reduce_fn(result, case): function
85 test_combinations = functools.reduce(reduce_fn, cases, [])
|
| D | map_fusion_test.py | 58 def reduce_fn(x, y): function
63 return functools.reduce(reduce_fn, cases, [])
|
| D | hoist_random_uniform_test.py | 59 def reduce_fn(x, y): function
65 return functools.reduce(reduce_fn, cases, [])
|
| D | filter_fusion_test.py | 53 def reduce_fn(x, y): function
59 return functools.reduce(reduce_fn, cases, [])
|
| D | map_and_filter_fusion_test.py | 68 def reduce_fn(x, y): function
74 return functools.reduce(reduce_fn, cases, [])
|
| D | map_parallelization_test.py | 49 def reduce_fn(x, y): function
55 return functools.reduce(reduce_fn, cases, [])
|
| D | map_vectorization_test.py | 53 def reduce_fn(x, y): function
57 return functools.reduce(reduce_fn, cases, [])
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops_test.py | 241 def reduce_fn(): function
254 got = reduce_fn()
258 got = def_function.function(reduce_fn)()
690 def reduce_fn(): function
701 self.assertAllEqual(reduce_fn(), [1. * num_replicas])
786 def reduce_fn(v): function
804 reduced = reduce_fn(v)
|
| D | input_lib.py | 936 def reduce(self, initial_state, reduce_fn): argument
949 state = reduce_fn(state, data)
|
| D | input_lib_test.py | 1115 def sum_while_loop(iterator, reduce_fn): argument
1119 sums = reduce_fn(sums, iterator)
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure.py | 335 def reduce_fn(state, value): function
340 reduce_fn, zip(nest.flatten(element_spec), nest.flatten(element)), [])
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_gradients_test.py | 941 def reduce_fn(x): function
950 g, _ = reduce_fn(constant_op.constant([7.0]))
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 340 XlaComputation reduce_fn = b->BuildAndNoteError(); in XLA_TEST_P() local
345 reduce_fn, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 632 def reduce_fn(input_tensor, axis, keepdims): function
655 reduce_fn = math_ops.reduce_variance
658 reduce_fn,
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 222 def reduce_fn(p, q): function
231 reduced = pfor_config.reduce(reduce_fn, x_i, y_i)
235 ans = reduce_fn(x, y) + x
|