| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reduction_ops_test_big.py | 32 def _tf_reduce(self, x, reduction_axes, keepdims): argument
39 def _tf_reduce_max(self, x, reduction_axes, keepdims): argument
42 def _tf_reduce_all(self, x, reduction_axes, keepdims): argument
45 def _tf_reduce_mean(self, x, reduction_axes, keepdims): argument
48 def _tf_reduce_sum(self, x, reduction_axes, keepdims): argument
|
| D | reduction_ops_test.py | 139 def _tf_reduce(self, x, reduction_axes, keepdims): argument
142 def _np_reduce(self, x, reduction_axes, keepdims): argument
157 def _compare(self, x, reduction_axes, keepdims, feed_dict=None): argument
165 def _compareAll(self, x, reduction_axes, feed_dict=None): argument
177 def _compareGradient(self, x, reduction_axes, rtol=1e-8, atol=1e-8): argument
200 def _tf_reduce(self, x, reduction_axes, keepdims): argument
203 def _np_reduce(self, x, reduction_axes, keepdims): argument
425 def _tf_reduce(self, x, reduction_axes, keepdims): argument
428 def _np_reduce(self, x, reduction_axes, keepdims): argument
547 def _tf_reduce(self, x, reduction_axes, keepdims): argument
[all …]
|
| D | sparse_ops_test.py | 631 def _compare(self, sp_t, reduction_axes, ndims, keep_dims, do_sum): argument
677 def _compare_all(self, sp_t, reduction_axes, ndims): argument
747 def _testSparseReduceShape(self, sp_t, reduction_axes, ndims, keep_dims, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 1339 reduction_axes=None, keep_dims=None): argument
1424 reduction_axes=None, argument
1557 reduction_axes=None, keep_dims=None): argument
1625 reduction_axes=None, argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 713 def _calculate_mean_and_var(self, inputs, reduction_axes, keep_dims): argument
716 def _moments(self, inputs, reduction_axes, keep_dims): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_reduce_op.cc | 59 std::vector<int32> reduction_axes(axes_slice.begin(), axes_slice.end()); in SparseTensorReduceHelper() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 2873 reduction_axes, argument
2896 reduction_axes, argument
2939 reduction_axes, argument
2974 def normalize_batch_in_training(x, gamma, beta, reduction_axes, epsilon=1e-3): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | quantize_training.cc | 476 Node* reduction_axes; in MakeEMAMinMaxVars() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 628 const Output& reduction_axes) { in ReducedShapeHelper()
|