| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | reduce_test.py | 50 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
|
| D | optional_test.py | 67 def reduce_fn(x, y): function
108 def reduce_fn(x, y): function
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | sparse_test.py | 51 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
|
| D | structure_test.py | 85 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
|
| D | random_seed_test.py | 94 def reduce_fn(x, y): function
|
| D | structure.py | 340 def reduce_fn(state, value): function
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | group_by_reducer_test.py | 53 def reduce_fn(x, y): function
110 def reduce_fn(x, _): function
178 def reduce_fn(state, value): function
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ReduceOps.h | 18 using reduce_fn = void(*)(TensorIterator &); variable
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | optimization_test.py | 68 def reduce_fn(key, bucket): function
92 def reduce_fn(x, y): function
|
| D | noop_elimination_test.py | 77 def reduce_fn(result, case): function
|
| D | map_fusion_test.py | 55 def reduce_fn(x, y): function
|
| D | filter_fusion_test.py | 50 def reduce_fn(x, y): function
|
| D | map_parallelization_test.py | 46 def reduce_fn(x, y): function
|
| D | map_and_filter_fusion_test.py | 65 def reduce_fn(x, y): function
|
| /external/pytorch/torch/ao/pruning/_experimental/activation_sparsifier/ |
| D | activation_sparsifier.py | 94 reduce_fn=None, argument
189 reduce_fn=None, argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops_test.py | 239 def reduce_fn(): function
899 def reduce_fn(): function
1017 def reduce_fn(v): function
|
| D | input_lib_test.py | 1179 def sum_while_loop(iterator, reduce_fn): argument
|
| /external/pytorch/torch/distributed/pipelining/ |
| D | microbatch.py | 40 def __init__(self, init_value, reduce_fn): argument
|
| /external/pytorch/test/ao/sparsity/ |
| D | test_activation_sparsifier.py | 297 def reduce_fn(x): function
|
| /external/pytorch/torch/masked/maskedtensor/ |
| D | reductions.py | 113 def reduce_fn(*args, **kwargs): function
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| D | dynamic_sharding_test.py | 97 def reduce_fn(_, window): function
|
| /external/executorch/kernels/portable/cpu/util/ |
| D | kernel_ops_util.h | 141 const ReduceOp& reduce_fn, in kernel_reduction_then_map_2d()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 1548 def reduce_fn(t): function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_gradients_test.py | 938 def reduce_fn(x): function
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 629 def reduce_fn(input_tensor, axis, keepdims): function
|