| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| H A D | mean.mod.py | 4 keepDims = Int32Scalar("keepDims", 0) variable
7 model = model.Operation("MEAN", i1, axis, keepDims).To(output)
|
| H A D | mean_float_1.mod.py | 4 keepDims = Int32Scalar("keepDims", 0) variable
7 model = model.Operation("MEAN", i1, axis, keepDims).To(output)
|
| H A D | mean_float_1_relaxed.mod.py | 20 keepDims = Int32Scalar("keepDims", 0) variable
23 model = model.Operation("MEAN", i1, axis, keepDims).To(output)
|
| H A D | mean_float_2.mod.py | 4 keepDims = Int32Scalar("keepDims", 1) variable
7 model = model.Operation("MEAN", i1, axis, keepDims).To(output)
|
| H A D | mean_float_2_relaxed.mod.py | 20 keepDims = Int32Scalar("keepDims", 1) variable
23 model = model.Operation("MEAN", i1, axis, keepDims).To(output)
|
| H A D | mean_quant8_1.mod.py | 4 keepDims = Int32Scalar("keepDims", 0) variable
7 model = model.Operation("MEAN", i1, axis, keepDims).To(output)
|
| H A D | mean_quant8_2.mod.py | 4 keepDims = Int32Scalar("keepDims", 1) variable
7 model = model.Operation("MEAN", i1, axis, keepDims).To(output)
|
| H A D | mean_relaxed.mod.py | 20 keepDims = Int32Scalar("keepDims", 0) variable
23 model = model.Operation("MEAN", i1, axis, keepDims).To(output)
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| H A D | mean.model.cpp | 10 auto keepDims = model->addOperand(&type2); local
16 model->setOperandValue(keepDims, keepDims_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_MEAN, {input, axis, keepDims}, {output});
|
| H A D | mean_float_1.model.cpp | 10 auto keepDims = model->addOperand(&type2); local
16 model->setOperandValue(keepDims, keepDims_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_MEAN, {input, axis, keepDims}, {output});
|
| H A D | mean_float_1_relaxed.model.cpp | 10 auto keepDims = model->addOperand(&type2); local
16 model->setOperandValue(keepDims, keepDims_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_MEAN, {input, axis, keepDims}, {output});
|
| H A D | mean_float_2.model.cpp | 10 auto keepDims = model->addOperand(&type2); local
16 model->setOperandValue(keepDims, keepDims_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_MEAN, {input, axis, keepDims}, {output});
|
| H A D | mean_float_2_relaxed.model.cpp | 10 auto keepDims = model->addOperand(&type2); local
16 model->setOperandValue(keepDims, keepDims_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_MEAN, {input, axis, keepDims}, {output});
|
| H A D | mean_quant8_1.model.cpp | 10 auto keepDims = model->addOperand(&type2); local
16 model->setOperandValue(keepDims, keepDims_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_MEAN, {input, axis, keepDims}, {output});
|
| H A D | mean_quant8_2.model.cpp | 10 auto keepDims = model->addOperand(&type2); local
16 model->setOperandValue(keepDims, keepDims_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_MEAN, {input, axis, keepDims}, {output});
|
| H A D | mean_relaxed.model.cpp | 10 auto keepDims = model->addOperand(&type2); local
16 model->setOperandValue(keepDims, keepDims_init, sizeof(int32_t) * 1);
17 model->addOperation(ANEURALNETWORKS_MEAN, {input, axis, keepDims}, {output});
|
| /frameworks/ml/nn/common/operations/ |
| H A D | SimpleMath.cpp | 280 const int32_t* axis, const Shape& axisShape, bool keepDims,
298 axis, axisSize, keepDims, scratchBuffer, resolvedAxis); local
307 axis, axisSize, keepDims, scratchBuffer, resolvedAxis); local
279 meanGeneric(const uint8_t* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, bool keepDims, uint8_t* outputData, const Shape& outputShape) argument
|
| /frameworks/ml/nn/common/ |
| H A D | CpuExecutor.cpp | 1505 int32_t keepDims = getScalarData<int32_t>(mOperands[ins[2]]); local
1513 keepDims > 0,
1520 keepDims > 0,
|
| H A D | OperationsUtils.cpp | 751 bool keepDims,
762 if (keepDims) {
748 meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, Shape* output) argument
|