Searched refs:inputDim (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | MirrorPad.cpp | 200 const auto inputDim = getSizeOfDimension(inputShape, i); in eval() local
201 NN_RET_CHECK(checkAsInt(inputDim)); in eval()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3601 uint32_t inputDim[] = {2, 2, 2, 2}; in normalizationOpTest() local
3602 OperationTestBase normalizationTest(operationCode, {getOpType(operandCode, 4, inputDim)}, in normalizationOpTest()
3603 {getOpType(operandCode, 4, inputDim)}); in normalizationOpTest()
3607 operationCode, {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32)}, in normalizationOpTest()
3608 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4)}}); in normalizationOpTest()
3633 uint32_t inputDim[] = {2, 2, 2, 6}; in localResponseNormOpTest() local
3636 {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32), in localResponseNormOpTest()
3638 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4), {0}}}); in localResponseNormOpTest()
3643 {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32), in localResponseNormOpTest()
3646 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4), {0}}}); in localResponseNormOpTest()
[all …]
|