| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | mean_quant8_signed.mod.py | 19 keepDims = Int32Scalar("keepDims", 0) variable
43 keepDims = Int32Scalar("keepDims", 1) variable
|
| /frameworks/ml/nn/common/operations/ |
| D | SimpleMath.cpp | 34 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
49 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
|
| D | Reduce.cpp | 147 bool keepDims = context->getInputValue<bool8>(kInputKeepDims); in prepare() local
|
| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| D | mean.mod.py | 4 keepDims = Int32Scalar("keepDims", 0) variable
|
| D | mean_quant8_2.mod.py | 4 keepDims = Int32Scalar("keepDims", 1) variable
|
| D | mean_quant8_1.mod.py | 4 keepDims = Int32Scalar("keepDims", 0) variable
|
| D | mean_float_2.mod.py | 4 keepDims = Int32Scalar("keepDims", 1) variable
|
| D | mean_float_1.mod.py | 4 keepDims = Int32Scalar("keepDims", 0) variable
|
| D | mean_relaxed.mod.py | 20 keepDims = Int32Scalar("keepDims", 0) variable
|
| D | mean_float_2_relaxed.mod.py | 20 keepDims = Int32Scalar("keepDims", 1) variable
|
| D | mean_float_1_relaxed.mod.py | 20 keepDims = Int32Scalar("keepDims", 0) variable
|
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | mean_float16.mod.py | 4 keepDims = Int32Scalar("keepDims", 0) variable
|
| /frameworks/ml/nn/runtime/test/fuzzing/operation_signatures/ |
| D | Reduce.cpp | 40 bool keepDims; in reduceOpConstructor() local
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 594 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare()
|
| D | CpuExecutor.cpp | 1360 int32_t keepDims = getScalarData<int32_t>(operands[ins[2]]); in executeOperation() local
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1720 ANeuralNetworksOperandType keepDims = getOpType(ANEURALNETWORKS_INT32); in meanOpTest() local
|