Searched refs:reduce_fn (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | reduce_test.py | 48 def reduce_fn(state, value): function
55 result = ds.reduce(constant_op.constant(0, dtype=dtypes.int64), reduce_fn)
60 def reduce_fn(state, value): function
68 reduce_fn)
78 def reduce_fn(state, _): function
83 result = ds.reduce(np.int64(0), reduce_fn)
90 def reduce_fn(_, value): function
101 result = ds.reduce(make_sparse_fn(0), reduce_fn)
106 def reduce_fn(state, value): function
123 result = ds.reduce(map_fn(0), reduce_fn)
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | group_by_reducer_test.py | 54 def reduce_fn(x, y): function
60 reduce_func=reduce_fn,
108 def reduce_fn(x, _): function
117 reduce_func=reduce_fn,
176 def reduce_fn(state, value): function
184 reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| D | reduce_dataset_test.py | 46 def reduce_fn(state, value): function
52 sum_reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | optimize_dataset_test.py | 78 def reduce_fn(key, bucket): function
83 grouping.group_by_window(lambda _: 0, reduce_fn, 10))
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_gradients_test.py | 740 def reduce_fn(x): function
749 g, _ = reduce_fn(constant_op.constant([7.0]))
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 880 def define_reduce_op(op_name, reduce_fn): argument
955 reduce_op = reduce_fn(
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 317 XlaComputation reduce_fn = b->BuildAndNoteError(); in XLA_TEST_P() local
322 reduce_fn, in XLA_TEST_P()
|