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