| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| reduce_test.py | 48 def reduce_fn(state, value): function in function:ReduceTest.testSumTuple
55 result = ds.reduce(constant_op.constant(0, dtype=dtypes.int64), reduce_fn)
60 def reduce_fn(state, value): function in function:ReduceTest.testSumAndCount
68 reduce_fn)
78 def reduce_fn(state, _): function in function:ReduceTest.testSkipEagerSquareUsingPlaceholder
83 result = ds.reduce(np.int64(0), reduce_fn)
90 def reduce_fn(_, value): function in function:ReduceTest.testSparse
101 result = ds.reduce(make_sparse_fn(0), reduce_fn)
106 def reduce_fn(state, value): function in function:ReduceTest.testNested
123 result = ds.reduce(map_fn(0), reduce_fn)
138 def reduce_fn(state, value): function in function:ReduceTest.testDatasetSideEffect
156 def reduce_fn(state, value): function in function:ReduceTest.testSideEffect
[all...] |
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| reduce_dataset_test.py | 46 def reduce_fn(state, value): function in function:ReduceDatasetTest.testReduceDataset
52 sum_reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| group_by_reducer_test.py | 54 def reduce_fn(x, y): function in function:GroupByReducerTest.testAverage
60 reduce_func=reduce_fn,
108 def reduce_fn(x, _): function in function:GroupByReducerTest.testChangingStateShape
117 reduce_func=reduce_fn,
176 def reduce_fn(state, value): function in function:GroupByReducerTest.testTuple
184 reducer = grouping.Reducer(init_fn, reduce_fn, finalize_fn)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| optimize_dataset_test.py | 78 def reduce_fn(key, bucket): function in function:_generate_captured_refvar_test_cases.make_group_by_window_dataset
83 grouping.group_by_window(lambda _: 0, reduce_fn, 10))
|
| /external/tensorflow/tensorflow/python/eager/ |
| function_gradients_test.py | 740 def reduce_fn(x): function in function:FunctionGradientsTest.testBackwardNone
749 g, _ = reduce_fn(constant_op.constant([7.0]))
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reduce_window_test.cc | 317 XlaComputation reduce_fn = b->BuildAndNoteError(); local
322 reduce_fn,
[all...] |
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| ops.py | 880 def define_reduce_op(op_name, reduce_fn):
885 reduce_fn: function to call to evaluate the op on a tf.Tensor.
[all...] |
