Searched refs:inputDim (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3563 uint32_t inputDim[] = {2, 2, 2, 2}; in normalizationOpTest() local
3564 OperationTestBase normalizationTest(operationCode, {getOpType(operandCode, 4, inputDim)}, in normalizationOpTest()
3565 {getOpType(operandCode, 4, inputDim)}); in normalizationOpTest()
3569 operationCode, {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32)}, in normalizationOpTest()
3570 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4)}}); in normalizationOpTest()
3595 uint32_t inputDim[] = {2, 2, 2, 6}; in localResponseNormOpTest() local
3598 {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32), in localResponseNormOpTest()
3600 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4), {0}}}); in localResponseNormOpTest()
3605 {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32), in localResponseNormOpTest()
3608 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4), {0}}}); in localResponseNormOpTest()
[all …]
|