Home
last modified time | relevance | path

Searched defs:keepDims (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/android-nn-driver/test/1.1/
H A DMean.cpp26 int32_t keepDims, in MeanTestImpl()
110 int32_t keepDims = 0; variable
121 int32_t keepDims = 1; variable
135 int32_t keepDims = 0; variable
146 int32_t keepDims = 1; variable
165 int32_t keepDims = 0; variable
176 int32_t keepDims = 1; variable
190 int32_t keepDims = 0; variable
201 int32_t keepDims = 1; variable
218 int32_t keepDims = 0; variable
[all …]
/aosp_15_r20/external/android-nn-driver/test/1.2/
H A DMean.cpp28 int32_t keepDims, in MeanTestImpl()
115 int32_t keepDims = 0; variable
128 int32_t keepDims = 1; variable
150 int32_t keepDims = 0; variable
165 int32_t keepDims = 1; variable
185 int32_t keepDims = 0; variable
198 int32_t keepDims = 1; variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dmean_quant8_signed.mod.py19 keepDims = Int32Scalar("keepDims", 0) variable
43 keepDims = Int32Scalar("keepDims", 1) variable
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DSimpleMath.cpp40 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
55 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
DReduce.cpp84 bool keepDims = context->getInputValue<bool8>(kInputKeepDims); in prepare() local
/aosp_15_r20/external/armnn/delegate/test/
H A DReduceTestHelper.hpp30 const bool keepDims, in CreateReduceTfLiteModel()
157 const bool keepDims, in ReduceTest()
H A DControlTestHelper.hpp121 const bool keepDims, in CreateMeanTfLiteModel()
269 const bool keepDims, in MeanTest()
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/
Dmean_quant8_2.mod.py4 keepDims = Int32Scalar("keepDims", 1) variable
Dmean_float_1.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
Dmean_quant8_1.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
Dmean.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
Dmean_float_2.mod.py4 keepDims = Int32Scalar("keepDims", 1) variable
Dmean_relaxed.mod.py20 keepDims = Int32Scalar("keepDims", 0) variable
Dmean_float_2_relaxed.mod.py20 keepDims = Int32Scalar("keepDims", 1) variable
Dmean_float_1_relaxed.mod.py20 keepDims = Int32Scalar("keepDims", 0) variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dmean_float16.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
/aosp_15_r20/external/armnn/src/armnnUtils/test/
H A DParserHelperTest.cpp21 bool keepDims = false; variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/
DReduce.cpp40 bool keepDims; in reduceOpConstructor() local
/aosp_15_r20/external/armnn/src/armnnUtils/
H A DParserHelper.cpp54 bool keepDims, in CalculateReducedOutputTensoInfo()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DReduceSumTestImpl.cpp29 bool keepDims = false) in ReduceTestCommon()
H A DReduceProdTestImpl.cpp28 bool keepDims = false) in ReduceTestCommon()
H A DReductionTestImpl.cpp30 bool keepDims = false) in ReductionTestCommon()
H A DMeanTestImpl.hpp26 bool keepDims, in MeanTestHelper()
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A DArmComputeUtils.hpp354 const bool keepDims) in ComputeReductionTensorShape()
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp501 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare()

12