Searched refs:inputDim (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3562 uint32_t inputDim[] = {2, 2, 2, 2}; in normalizationOpTest() local
3563 OperationTestBase normalizationTest(operationCode, {getOpType(operandCode, 4, inputDim)}, in normalizationOpTest()
3564 {getOpType(operandCode, 4, inputDim)}); in normalizationOpTest()
3568 operationCode, {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32)}, in normalizationOpTest()
3569 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4)}}); in normalizationOpTest()
3594 uint32_t inputDim[] = {2, 2, 2, 6}; in localResponseNormOpTest() local
3597 {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32), in localResponseNormOpTest()
3599 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4), {0}}}); in localResponseNormOpTest()
3604 {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32), in localResponseNormOpTest()
3607 {getOpType(operandCode, 4, inputDim)}, {{TensorRankConstraint::UpTo(4), {0}}}); in localResponseNormOpTest()
[all …]
|