| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | reduction_ops_test.py | 135 def _tf_reduce(self, x, reduction_axes, keepdims): argument
138 def _np_reduce(self, x, reduction_axes, keepdims): argument
153 def _compare(self, x, reduction_axes, keepdims, feed_dict=None): argument
154 np_ans = self._np_reduce(x, reduction_axes, keepdims)
156 tf_ans = self._tf_reduce(x, reduction_axes, keepdims)
161 def _compareAll(self, x, reduction_axes, feed_dict=None): argument
162 if reduction_axes is not None and np.shape(reduction_axes) == (1,):
164 self._compareAll(x, reduction_axes[0])
165 self._compare(x, reduction_axes, keepdims=False, feed_dict=feed_dict)
166 self._compare(x, reduction_axes, keepdims=True, feed_dict=feed_dict)
[all …]
|
| D | reduction_ops_test_big.py | 28 def _tf_reduce(self, x, reduction_axes, keepdims): argument
35 def _tf_reduce_max(self, x, reduction_axes, keepdims): argument
36 return math_ops.reduce_max(x, reduction_axes, keepdims)
38 def _tf_reduce_all(self, x, reduction_axes, keepdims): argument
39 return math_ops.reduce_all(x, reduction_axes, keepdims)
41 def _tf_reduce_mean(self, x, reduction_axes, keepdims): argument
42 return math_ops.reduce_mean(x, reduction_axes, keepdims)
44 def _tf_reduce_sum(self, x, reduction_axes, keepdims): argument
45 return math_ops.reduce_sum(x, reduction_axes, keepdims)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops.h | 55 const ReductionAxes& reduction_axes, const Reducer& reducer) {
56 out.device(d) = in.reduce(reduction_axes, reducer);
68 const ReductionAxes& reduction_axes, \
75 in_as_intermediate.reduce(reduction_axes, intermediate_reducer) \
88 const ReductionAxes& reduction_axes,
92 out.device(d) = in.reduce(reduction_axes, sum_reducer) /
105 const ReductionAxes& reduction_axes, \
111 reduction_axes, sum_reducer) / \
132 const ReductionAxes& reduction_axes,
137 (in * in.conjugate()).reduce(reduction_axes, sum_reducer).sqrt();
[all …]
|
| D | reduction_gpu_kernels.cu.h | 1075 const ReductionAxes& reduction_axes, Op op) {
1082 reduction_axes[0] == 1) { // row reduction
1085 reduction_axes[0] == 0) { // column reduction
1087 } else if (in_rank == 3 && out_rank == 2 && reduction_axes[0] == 1) {
1096 } else if (in_rank == 3 && out_rank == 1 && reduction_axes[0] == 0 &&
1097 reduction_axes[1] == 2) {
1104 if (out_rank == 1) ss << " " << reduction_axes[0];
1105 if (out_rank == 2) ss << " " << reduction_axes[1];
1114 const ReductionAxes& reduction_axes,
1122 const ReductionAxes& reduction_axes,
[all …]
|
| D | sparse_reduce_op.cc | 61 std::vector<int32> reduction_axes(axes_slice.begin(), axes_slice.end()); in SparseTensorReduceHelper() local
63 for (int64_t i = 0; i < reduction_axes.size(); ++i) { in SparseTensorReduceHelper()
64 reduction_axes[i] = (reduction_axes[i] + ndims) % ndims; in SparseTensorReduceHelper()
66 std::sort(reduction_axes.begin(), reduction_axes.end()); in SparseTensorReduceHelper()
76 perm.begin(), perm.end(), reduction_axes.begin(), reduction_axes.end(), in SparseTensorReduceHelper()
|
| D | reduction_ops_common_gpu.h | 33 const ReductionAxes& reduction_axes,
|
| D | reduction_ops_gpu_bool.cu.cc | 40 const Eigen::array<Index, NUM_AXES>& reduction_axes, \
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_ops_test.py | 690 def _compare(self, sp_t, reduction_axes, ndims, keep_dims, do_sum): argument
694 if reduction_axes is None:
700 if not isinstance(reduction_axes, list): # Single scalar.
701 reduction_axes = [reduction_axes]
702 reduction_axes = np.array(reduction_axes).astype(np.int32)
704 reduction_axes = (reduction_axes + ndims) % ndims
706 reduction_axes.sort()
707 for ra in reduction_axes.ravel()[::-1]:
715 tf_dense_ans = sparse_ops.sparse_reduce_sum(sp_t, reduction_axes,
718 tf_dense_ans = sparse_ops.sparse_reduce_max(sp_t, reduction_axes,
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseReduceMaxSparse.pbtxt | 23 name: "reduction_axes"
40 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
42 `reduction_axes`. If `keep_dims` is true, the reduced dimensions are retained
45 If `reduction_axes` has no entries, all dimensions are reduced, and a tensor
|
| D | api_def_SparseReduceSumSparse.pbtxt | 23 name: "reduction_axes"
40 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
42 `reduction_axes`. If `keep_dims` is true, the reduced dimensions are retained
45 If `reduction_axes` has no entries, all dimensions are reduced, and a tensor
|
| D | api_def_SparseReduceSum.pbtxt | 23 name: "reduction_axes"
46 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
48 `reduction_axes`. If `keep_dims` is true, the reduced dimensions are retained
51 If `reduction_axes` has no entries, all dimensions are reduced, and a tensor
|
| D | api_def_SparseReduceMax.pbtxt | 23 name: "reduction_axes"
46 Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless
48 `reduction_axes`. If `keep_dims` is true, the reduced dimensions are retained
51 If `reduction_axes` has no entries, all dimensions are reduced, and a tensor
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SparseReduceMax.pbtxt | 16 name: "reduction_axes"
63 name: "reduction_axes"
112 name: "reduction_axes"
162 name: "reduction_axes"
|
| D | SparseReduceMaxSparse.pbtxt | 16 name: "reduction_axes"
71 name: "reduction_axes"
128 name: "reduction_axes"
186 name: "reduction_axes"
|
| D | SparseReduceSum.pbtxt | 16 name: "reduction_axes"
68 name: "reduction_axes"
122 name: "reduction_axes"
177 name: "reduction_axes"
|
| D | SparseReduceSumSparse.pbtxt | 16 name: "reduction_axes"
76 name: "reduction_axes"
138 name: "reduction_axes"
201 name: "reduction_axes"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseReduceMax.pbtxt | 16 name: "reduction_axes"
63 name: "reduction_axes"
112 name: "reduction_axes"
162 name: "reduction_axes"
|
| D | SparseReduceMaxSparse.pbtxt | 16 name: "reduction_axes"
71 name: "reduction_axes"
128 name: "reduction_axes"
186 name: "reduction_axes"
|
| D | SparseReduceSum.pbtxt | 16 name: "reduction_axes"
68 name: "reduction_axes"
122 name: "reduction_axes"
177 name: "reduction_axes"
|
| D | SparseReduceSumSparse.pbtxt | 16 name: "reduction_axes"
76 name: "reduction_axes"
138 name: "reduction_axes"
201 name: "reduction_axes"
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | grad_helper.cc | 33 const Output& reduction_axes) { in ReducedShapeHelper() argument
52 auto axes = Mod(scope, Add(scope, reduction_axes, input_rank), input_rank); in ReducedShapeHelper()
|
| D | grad_helper.h | 32 const Output& reduction_axes);
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 86 _, reduction_axes = tf.raw_ops.BroadcastGradientArgs(
89 grad, axis=reduction_axes, keepdims=True)
150 _, reduction_axes = tf.raw_ops.BroadcastGradientArgs(
153 grad, axis=reduction_axes, keepdims=True)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.pbtxt | 73 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
77 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
81 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
85 …argspec: "args=[\'sp_input\', \'axis\', \'keepdims\', \'reduction_axes\', \'keep_dims\'], varargs=…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 1332 reduction_axes=None, keep_dims=None): argument
1401 reduction_axes)
1417 reduction_axes=None, argument
1451 reduction_axes)
1550 reduction_axes=None, keep_dims=None): argument
1602 reduction_axes)
1618 reduction_axes=None, argument
1652 reduction_axes)
|