| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_batchnorm_test.py | 354 def _npSuffStats(self, x, axes, shift, keep_dims): argument
370 def _opSuffStats(self, x, axes, shift, keep_dims): argument
373 def _testSuffStats(self, x_shape, axes, shift, keep_dims, has_shape): argument
457 def _unweighted_moments(self, x, axes, keep_dims=False, extra_out_grads=None): argument
467 def RunMomentTestWithDynamicShape(self, shape, axes, keep_dims, dtype): argument
494 def RunMomentTest(self, shape, axes, keep_dims, dtype): argument
619 def _unweighted_moments(self, x, axes, keep_dims=False, extra_out_grads=None): argument
626 def RunMomentTest(self, shape, axes, keep_dims, dtype, dynshapes=False): argument
652 def RunMomentTestWithDynamicShape(self, shape, axes, keep_dims, dtype): argument
659 keep_dims, argument
|
| D | math_ops.py | 1320 keep_dims=None): argument
1467 keep_dims=None, argument
1605 keep_dims=None): argument
1858 keep_dims=None): argument
1901 keep_dims=None): argument
1980 keep_dims=None): argument
2059 keep_dims=None): argument
2156 keep_dims=None): argument
2253 keep_dims=None): argument
|
| D | sparse_ops.py | 1107 reduction_axes=None, keep_dims=None): argument
1178 keep_dims=None): argument
1296 reduction_axes=None, keep_dims=None): argument
1354 keep_dims=None): argument
|
| D | linalg_ops.py | 505 keep_dims=None): argument
|
| D | nn_impl.py | 811 def sufficient_statistics(x, axes, shift=None, keep_dims=None, name=None, argument
933 keep_dims=None, argument
1029 def weighted_moments(x, axes, frequency_weights, name=None, keep_dims=None, argument
|
| D | string_ops.py | 330 keep_dims=False, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 62 std::initializer_list<int> axis, bool keep_dims) { in MeanOpConstModel()
76 const TensorData& axis, bool keep_dims) { in MeanOpDynamicModel()
91 std::initializer_list<int> axis, bool keep_dims) { in SumOpConstModel()
105 const TensorData& axis, bool keep_dims) { in SumOpDynamicModel()
120 std::initializer_list<int> axis, bool keep_dims) { in ProdOpConstModel()
134 const TensorData& axis, bool keep_dims) { in ProdOpDynamicModel()
149 std::initializer_list<int> axis, bool keep_dims) { in MaxOpConstModel()
163 const TensorData& axis, bool keep_dims) { in MaxOpDynamicModel()
178 std::initializer_list<int> axis, bool keep_dims) { in MinOpConstModel()
192 const TensorData& axis, bool keep_dims) { in MinOpDynamicModel()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_common.cc | 78 const bool keep_dims) { in Simplify()
|
| D | reduce_join_op.cc | 100 const TensorShape& input_shape, bool keep_dims) { in GetOutputShape()
|
| D | sparse_reduce_op.cc | 56 bool keep_dims) { in SparseTensorReduceHelper()
|
| /external/tensorflow/tensorflow/contrib/bayesflow/python/ops/ |
| D | monte_carlo_impl.py | 199 axis=0, keep_dims=False, name=None): argument
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | sample_stats.py | 223 keep_dims=False, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reduce_join_op_test.py | 105 keep_dims=False, argument
|
| D | sparse_ops_test.py | 614 def _compare(self, sp_t, reduction_axes, ndims, keep_dims, do_sum): argument
732 def _testSparseReduceShape(self, sp_t, reduction_axes, ndims, keep_dims, argument
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_unary.cc | 36 void ReduceGeneric(bool keep_dims, const std::vector<int>& axes, in ReduceGeneric()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 565 def _moments(self, inputs, reduction_axes, keep_dims): argument
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 1048 keep_dims=False, argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1004 bool keep_dims = Choose<bool>({false, true}); in TEST_F() local
1024 bool keep_dims = Choose<bool>({false, true}); in TEST_F() local
2290 bool keep_dims = Choose<bool>({false, true}); in TEST_F() local
2374 bool keep_dims = Choose<bool>({false, true}); in TEST_F() local
2388 bool keep_dims = Choose<bool>({false, true}); in TEST_F() local
2553 bool keep_dims = Choose<bool>({false, true}); in TEST_F() local
3193 bool keep_dims = Choose<bool>({false, true}); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | util_test.py | 811 def _reduce_weighted_logsumexp(self, logx, w, axis, keep_dims=False): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1166 bool keep_dims; in ReductionShape() local
1636 bool keep_dims = false; in SparseReduceShapeFn() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 138 bool keep_dims) { in Reduce()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 956 int32_t keep_dims = 0; in Map() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 288 bool keep_dims; member
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 698 bool keep_dims = false; in RegisterNodeWithRank() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2511 const NodeDef& node, const TensorShapeProto& input_shape, bool keep_dims, in IsReductionSimplifiableToIdentity()
2568 bool keep_dims = in SimplifyReduction() local
|