Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dmean_quant8_signed.mod.py19 keepDims = Int32Scalar("keepDims", 0) variable
43 keepDims = Int32Scalar("keepDims", 1) variable
/packages/modules/NeuralNetworks/common/operations/
DSimpleMath.cpp34 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
49 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
DReduce.cpp149 bool keepDims = context->getInputValue<bool8>(kInputKeepDims); in prepare() local
/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/
Dmean.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
Dmean_quant8_2.mod.py4 keepDims = Int32Scalar("keepDims", 1) variable
Dmean_float_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_float_2_relaxed.mod.py20 keepDims = Int32Scalar("keepDims", 1) variable
Dmean_float_1_relaxed.mod.py20 keepDims = Int32Scalar("keepDims", 0) variable
Dmean_relaxed.mod.py20 keepDims = Int32Scalar("keepDims", 0) variable
/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dmean_float16.mod.py4 keepDims = Int32Scalar("keepDims", 0) variable
/packages/modules/NeuralNetworks/runtime/test/fuzzing/operation_signatures/
DReduce.cpp40 bool keepDims; in reduceOpConstructor() local
/packages/modules/NeuralNetworks/common/
DOperationsUtils.cpp592 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare()
DCpuExecutor.cpp1304 int32_t keepDims = getScalarData<int32_t>(operands[ins[2]]); in executeOperation() local
/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp1719 ANeuralNetworksOperandType keepDims = getOpType(ANEURALNETWORKS_INT32); in meanOpTest() local