| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ReduceMeanFixture.h | 51 …void setup(TensorShape shape, DataType data_type, Coordinates axis, bool keep_dims, QuantizationIn… in setup()
84 …target(TensorShape &src_shape, DataType data_type, Coordinates axis, bool keep_dims, QuantizationI… in compute_target()
114 …erence(TensorShape &src_shape, DataType data_type, Coordinates axis, bool keep_dims, QuantizationI… in compute_reference()
159 …void setup(TensorShape shape, DataType data_type, Coordinates axis, bool keep_dims, QuantizationIn… in setup()
170 void setup(TensorShape shape, DataType data_type, Coordinates axis, bool keep_dims) in setup()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | reduce_spmd_expander.cc | 96 llvm::SmallVector<int64_t, 4>* reduced_dims, bool* keep_dims, in ExtractDims()
112 bool* keep_dims, bool* matched) { in ExtractDims()
138 bool& keep_dims) { in ExtractReductionParameters()
173 bool keep_dims; in ComputeResultLayout() local
294 bool keep_dims; in ComputeLayoutBackward() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLReduceMean.cpp | 41 Status validate_config(const ITensorInfo *input, const Coordinates &reduction_axis, bool keep_dims,… in validate_config()
109 void CLReduceMean::configure(ICLTensor *input, const Coordinates &reduction_axis, bool keep_dims, I… in configure()
114 …ompile_context, ICLTensor *input, const Coordinates &reduction_axis, bool keep_dims, ICLTensor *ou… in configure()
198 …alidate(const ITensorInfo *input, const Coordinates &reduction_axis, bool keep_dims, const ITensor… in validate()
|
| D | CLReductionOperation.cpp | 48 …orInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op, bool keep_dims) in validate()
106 …gure(ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op, bool keep_dims) in configure()
111 …ext, ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op, bool keep_dims) in configure()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_batchnorm_test.py | 349 def _npSuffStats(self, x, axes, shift, keep_dims): argument
365 def _opSuffStats(self, x, axes, shift, keep_dims): argument
368 def _testSuffStats(self, x_shape, axes, shift, keep_dims, has_shape): argument
452 def _unweighted_moments(self, x, axes, keep_dims=False, extra_out_grads=None): argument
462 def RunMomentTestWithDynamicShape(self, shape, axes, keep_dims, dtype): argument
489 def RunMomentTest(self, shape, axes, keep_dims, dtype): argument
614 def _unweighted_moments(self, x, axes, keep_dims=False, extra_out_grads=None): argument
621 def RunMomentTest(self, shape, axes, keep_dims, dtype, dynshapes=False): argument
647 def RunMomentTestWithDynamicShape(self, shape, axes, keep_dims, dtype): argument
654 keep_dims, argument
|
| D | sparse_ops.py | 1332 reduction_axes=None, keep_dims=None): argument
1418 keep_dims=None): argument
1550 reduction_axes=None, keep_dims=None): argument
1619 keep_dims=None): argument
|
| D | math_ops.py | 2179 keep_dims=None): argument
2386 keep_dims=None, argument
2525 keep_dims=None): argument
2818 keep_dims=None): argument
2877 keep_dims=None): argument
3005 keep_dims=None): argument
3130 keep_dims=None): argument
3236 keep_dims=None): argument
3342 keep_dims=None): argument
|
| D | nn_impl.py | 1183 def sufficient_statistics(x, axes, shift=None, keep_dims=None, name=None, argument
1326 keep_dims=None, argument
1424 def weighted_moments(x, axes, frequency_weights, name=None, keep_dims=None, argument
|
| D | linalg_ops.py | 638 keep_dims=None): argument
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEReduceMean.cpp | 37 Status validate_config(const ITensorInfo *input, const Coordinates &reduction_axis, bool keep_dims,… in validate_config()
98 …alidate(const ITensorInfo *input, const Coordinates &reduction_axis, bool keep_dims, const ITensor… in validate()
103 void NEReduceMean::configure(ITensor *input, const Coordinates &reduction_axis, bool keep_dims, ITe… in configure()
|
| D | NEReductionOperation.cpp | 66 …orInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op, bool keep_dims) in validate()
105 …onfigure(ITensor *input, ITensor *output, unsigned int axis, ReductionOperation op, bool keep_dims) in configure()
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | ReductionLayerNode.cpp | 35 ReductionLayerNode::ReductionLayerNode(ReductionOperation op, unsigned int axis, bool keep_dims) in ReductionLayerNode()
52 bool ReductionLayerNode::keep_dims() const in keep_dims() function in arm_compute::graph::ReductionLayerNode
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test_common.h | 76 std::initializer_list<int> axis, bool keep_dims) { in BaseConstOpModel()
95 const TensorData& axis, bool keep_dims) { in BaseDynamicOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 184 bool keep_dims, int* temp_index, int* resolved_axis, in ReduceGeneric()
220 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean()
384 const int num_axis_dimensions, bool keep_dims, in QuantizedMeanOrSum()
478 bool keep_dims, int* temp_index, in QuantizedReduceProd()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reduce_tester.h | 59 inline ReduceTester& KeepDims(bool keep_dims) { in KeepDims()
|
| D | fully_connected_tester.h | 68 inline FullyConnectedTester& KeepDims(bool keep_dims) { in KeepDims()
|
| D | quantized_reduce_tester.h | 61 inline QuantizedReduceTester& KeepDims(bool keep_dims) { in KeepDims()
|
| D | quantized_fully_connected_tester.h | 116 inline QuantizedFullyConnectedTester& 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 | 101 const TensorShape& input_shape, bool keep_dims) { in GetOutputShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_tensor_helper.cc | 63 BoolAttr keep_dims) { in InferReductionOpType()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | reduce_test.cc | 29 std::initializer_list<int> axis, bool keep_dims) { in ReduceOpModel()
|
| /external/tensorflow/tensorflow/python/kernel_tests/strings_ops/ |
| D | reduce_join_op_test.py | 101 keep_dims=False, argument
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_unary.cc | 38 void ReduceGeneric(bool keep_dims, const std::vector<int>& axes, in ReduceGeneric()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | reduce.h | 408 bool keep_dims, int* normalized_dims, in QuantizedMeanOrSum()
675 bool keep_dims, int* normalized_dims, in MeanGeneral()
732 const int* axis, const int num_axis_dimensions, bool keep_dims, in Mean()
|