Lines Matching refs:operandCode
1290 void simpleMathOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in simpleMathOpTest() argument
1292 ANeuralNetworksOperandType input1 = getOpType(operandCode, 4, inputDimensions); in simpleMathOpTest()
1399 void binaryOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in binaryOpTest() argument
1401 ANeuralNetworksOperandType input1 = getOpType(operandCode, 5, inputDimensions); in binaryOpTest()
1450 void activationOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in activationOpTest() argument
1452 ANeuralNetworksOperandType input = getOpType(operandCode, 4, inputDimensions); in activationOpTest()
1637 void eluOpTest(int32_t operandCode) { in eluOpTest() argument
1639 ANeuralNetworksOperandType input = getOpType(operandCode, 4, inputDimensions); in eluOpTest()
1640 ANeuralNetworksOperandType alpha = (operandCode == ANEURALNETWORKS_TENSOR_FLOAT32) in eluOpTest()
1792 void softmaxOpTest(int32_t operandCode) { in softmaxOpTest() argument
1794 ANeuralNetworksOperandType input = getOpType(operandCode, 4, inputDimensions); in softmaxOpTest()
1798 if (operandCode == ANEURALNETWORKS_TENSOR_FLOAT16) { in softmaxOpTest()
1828 void poolingOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in poolingOpTest() argument
1830 ANeuralNetworksOperandType input = getOpType(operandCode, 4, inputDimensions); in poolingOpTest()
1921 void spaceDepthOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in spaceDepthOpTest() argument
1923 ANeuralNetworksOperandType input = getOpType(operandCode, 4, inputDimensions); in spaceDepthOpTest()
1976 void spaceBatchOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in spaceBatchOpTest() argument
1978 ANeuralNetworksOperandType input = getOpType(operandCode, 4, inputDimensions); in spaceBatchOpTest()
2042 void transposeAndSqueezeOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in transposeAndSqueezeOpTest() argument
2044 ANeuralNetworksOperandType input = getOpType(operandCode, 4, inputDimensions); in transposeAndSqueezeOpTest()
2373 void fullyConnectedOpTest(int32_t operandCode) { in fullyConnectedOpTest() argument
2375 ANeuralNetworksOperandType input = getOpType(operandCode, 2, inputDimensions); in fullyConnectedOpTest()
2381 ANeuralNetworksOperandType bias = {.type = operandCode, in fullyConnectedOpTest()
2386 if (operandCode == ANEURALNETWORKS_TENSOR_QUANT8_ASYMM || in fullyConnectedOpTest()
2387 operandCode == ANEURALNETWORKS_TENSOR_QUANT8_ASYMM_SIGNED) { in fullyConnectedOpTest()
2422 void concatenationTest(int32_t operandCode) { in concatenationTest() argument
2424 ANeuralNetworksOperandType input1 = getOpType(operandCode, 2, inputDimensions); in concatenationTest()
2487 void embeddingLookupTest(int32_t operandCode) { in embeddingLookupTest() argument
2496 ANeuralNetworksOperandType input = getOpType(operandCode, 2, inputDimensions); in embeddingLookupTest()
2519 void hashtableLookupTest(int32_t operandCode) { in hashtableLookupTest() argument
2529 ANeuralNetworksOperandType values = getOpType(operandCode, 2, valuesDimensions); in hashtableLookupTest()
2553 void lshProjectionTest(int32_t operandCode, int32_t hashAndWeightOperandCode) { in lshProjectionTest() argument
2556 ANeuralNetworksOperandType input = getOpType(operandCode, 2, inputDimensions); in lshProjectionTest()
2676 void lstmTestV1_2(int32_t operandCode) { in lstmTestV1_2() argument
2677 SCOPED_TRACE(operandCode); in lstmTestV1_2()
2680 ANeuralNetworksOperandType floatTensor1D = {.type = operandCode, in lstmTestV1_2()
2685 ANeuralNetworksOperandType floatTensor2D = {.type = operandCode, in lstmTestV1_2()
2696 .type = (operandCode == ANEURALNETWORKS_TENSOR_FLOAT32) ? ANEURALNETWORKS_FLOAT32 in lstmTestV1_2()
2773 void lstmBidirectionalSequence(int32_t operandCode) { in lstmBidirectionalSequence() argument
2778 .type = operandCode, in lstmBidirectionalSequence()
2785 .type = operandCode, in lstmBidirectionalSequence()
2792 .type = operandCode, in lstmBidirectionalSequence()
2806 .type = operandCode == ANEURALNETWORKS_TENSOR_FLOAT32 ? ANEURALNETWORKS_FLOAT32 in lstmBidirectionalSequence()
2961 void randomMultinomialOpTest(int32_t operandCode) { in randomMultinomialOpTest() argument
2963 ANeuralNetworksOperandType input = {.type = operandCode, in randomMultinomialOpTest()
3143 void stridedSliceOpTest(int32_t operandCode) { in stridedSliceOpTest() argument
3145 ANeuralNetworksOperandType input = getOpType(operandCode, 2, inputDimensions); in stridedSliceOpTest()
3462 void channelShuffleOpTest(int32_t operandCode) { in channelShuffleOpTest() argument
3466 {getOpType(operandCode, 2, inoutDim), getOpType(ANEURALNETWORKS_INT32), in channelShuffleOpTest()
3468 {getOpType(operandCode, 2, inoutDim)}, {{TensorRankConstraint::UpTo(4)}}); in channelShuffleOpTest()
3536 void preluOpTest(int32_t operandCode) { in preluOpTest() argument
3540 {getOpType(operandCode, 4, inoutDim), getOpType(operandCode, 3, alphaDim)}, in preluOpTest()
3541 {getOpType(operandCode, 4, inoutDim)}); in preluOpTest()
3561 void normalizationOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in normalizationOpTest() argument
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()
3590 void localResponseNormOpTest(int32_t operandCode) { in localResponseNormOpTest() argument
3591 int32_t floatScalarType = (operandCode == ANEURALNETWORKS_TENSOR_FLOAT32) in localResponseNormOpTest()
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()
3649 void sliceTest(int32_t operandCode) { in sliceTest() argument
3656 {getOpType(operandCode, 3, inputDim), in sliceTest()
3659 {getOpType(operandCode, 3, outputDim)}); in sliceTest()