| /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 | 1947 keep_dims=None): argument
2154 keep_dims=None, argument
2293 keep_dims=None): argument
2583 keep_dims=None): argument
2642 keep_dims=None): argument
2770 keep_dims=None): argument
2895 keep_dims=None): argument
3001 keep_dims=None): argument
3107 keep_dims=None): argument
|
| D | sparse_ops.py | 1339 reduction_axes=None, keep_dims=None): argument
1425 keep_dims=None): argument
1557 reduction_axes=None, keep_dims=None): argument
1626 keep_dims=None): argument
|
| D | nn_impl.py | 1168 def sufficient_statistics(x, axes, shift=None, keep_dims=None, name=None, argument
1311 keep_dims=None, argument
1409 def weighted_moments(x, axes, frequency_weights, name=None, keep_dims=None, argument
|
| D | linalg_ops.py | 641 keep_dims=None): argument
|
| D | string_ops.py | 311 keep_dims=None, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 84 std::initializer_list<int> axis, bool keep_dims) { in MeanOpConstModel()
98 const TensorData& axis, bool keep_dims) { in MeanOpDynamicModel()
113 std::initializer_list<int> axis, bool keep_dims) { in SumOpConstModel()
127 const TensorData& axis, bool keep_dims) { in SumOpDynamicModel()
142 std::initializer_list<int> axis, bool keep_dims) { in ProdOpConstModel()
156 const TensorData& axis, bool keep_dims) { in ProdOpDynamicModel()
171 std::initializer_list<int> axis, bool keep_dims) { in MaxOpConstModel()
185 const TensorData& axis, bool keep_dims) { in MaxOpDynamicModel()
200 std::initializer_list<int> axis, bool keep_dims) { in MinOpConstModel()
214 const TensorData& axis, bool keep_dims) { in MinOpDynamicModel()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | fully_connected_tester.h | 67 inline FullyConnectedTester& KeepDims(bool keep_dims) { in KeepDims()
|
| D | reduce_tester.h | 59 inline ReduceTester& KeepDims(bool keep_dims) { in KeepDims()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_common.cc | 83 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/lite/kernels/internal/reference/ |
| D | reduce.h | 132 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric()
165 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean()
329 const int num_axis_dimensions, bool keep_dims, in QuantizedMeanOrSum()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization_v2.py | 138 def _calculate_mean_and_var(self, x, axes, keep_dims): argument
|
| 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/lite/delegates/hexagon/builders/tests/ |
| D | reduce_test.cc | 27 std::initializer_list<int> axis, bool keep_dims) { in ReduceOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1042 bool keep_dims = false; in matchAndRewrite() local
1068 bool keep_dims = false; in matchAndRewrite() local
1094 bool keep_dims = false; in matchAndRewrite() local
1120 bool keep_dims = false; in matchAndRewrite() local
1146 bool keep_dims = false; in matchAndRewrite() local
1172 bool keep_dims = false; in matchAndRewrite() local
1198 bool keep_dims = false; in matchAndRewrite() local
|
| D | legalize_common.cc | 2311 Value input_value, ElementsAttr axes_elems, bool keep_dims, in convertReduceOpCommon()
2373 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceAllOp()
2386 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceAnyOp()
2399 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceMinOp()
2412 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceMaxOp()
2425 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceProdOp()
2450 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceSumOp()
2500 Value input_value, ElementsAttr axes_elems, bool keep_dims) { in convertReduceMeanOp()
|
| D | legalize_tfl.cc | 1744 bool keep_dims = false; in matchAndRewrite() local
1770 bool keep_dims = false; in matchAndRewrite() local
1796 bool keep_dims = false; in matchAndRewrite() local
1822 bool keep_dims = false; in matchAndRewrite() local
1848 bool keep_dims = false; in matchAndRewrite() local
1874 bool keep_dims = false; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reduce_join_op_test.py | 105 keep_dims=False, argument
|
| D | sparse_ops_test.py | 631 def _compare(self, sp_t, reduction_axes, ndims, keep_dims, do_sum): argument
747 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/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 67 BoolAttr keep_dims; member
|
| /external/tensorflow/tensorflow/compiler/mlir/python/mlir_wrapper/ |
| D | ops.cc | 99 bool keep_dims = false) -> mlir::Operation* { in init_ops()
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 1053 keep_dims=False, argument
|