Lines Matching refs:operationCode
1039 void argMinMaxTest(ANeuralNetworksOperationType operationCode, int32_t inputOperandType) { in argMinMaxTest() argument
1054 OperationTestBase test(operationCode, {input0, axis}, {output}); in argMinMaxTest()
1301 void simpleMathOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in simpleMathOpTest() argument
1314 operationCode, {input1, input2, activation}, {output}, in simpleMathOpTest()
1410 void binaryOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in binaryOpTest() argument
1417 OperationTestBase test(operationCode, {input1, input2}, {output}); in binaryOpTest()
1461 void activationOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in activationOpTest() argument
1467 if (operationCode == ANEURALNETWORKS_FLOOR || operationCode == ANEURALNETWORKS_LOGISTIC || in activationOpTest()
1468 operationCode == ANEURALNETWORKS_RELU || operationCode == ANEURALNETWORKS_RELU1 || in activationOpTest()
1469 operationCode == ANEURALNETWORKS_RELU6 || operationCode == ANEURALNETWORKS_TANH) { in activationOpTest()
1472 OperationTestBase test(operationCode, {input}, {output}, inputRankMutators); in activationOpTest()
1763 void padOpTest(ANeuralNetworksOperationType operationCode, int32_t inputOperandCode) { in padOpTest() argument
1772 if (operationCode == ANEURALNETWORKS_MIRROR_PAD) { in padOpTest()
1780 if (operationCode == ANEURALNETWORKS_PAD) { in padOpTest()
1784 OperationTestBase test(operationCode, inputs, {output}, inputRankMutators); in padOpTest()
1867 void poolingOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in poolingOpTest() argument
1887 OperationTestBase explicitPoolingTest(operationCode, in poolingOpTest()
1895 operationCode, in poolingOpTest()
1907 operationCode, in poolingOpTest()
1913 OperationTestBase implicitNchwPoolingTest(operationCode, in poolingOpTest()
1960 void spaceDepthOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in spaceDepthOpTest() argument
1971 OperationTestBase spaceDepthTest(operationCode, {input, block_size}, {output}); in spaceDepthOpTest()
1979 OperationTestBase spaceDepthNchwTest(operationCode, {input, block_size, layout}, {output}); in spaceDepthOpTest()
2015 void spaceBatchOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in spaceBatchOpTest() argument
2033 if (operationCode == ANEURALNETWORKS_SPACE_TO_BATCH_ND) { in spaceBatchOpTest()
2034 OperationTestBase spaceBatchTest(operationCode, {input, blockShape, padding}, {output}); in spaceBatchOpTest()
2037 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, padding, layout}, in spaceBatchOpTest()
2041 OperationTestBase spaceBatchTest(operationCode, {input, blockShape}, {output}); in spaceBatchOpTest()
2044 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, layout}, {output}); in spaceBatchOpTest()
2081 void transposeAndSqueezeOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in transposeAndSqueezeOpTest() argument
2093 OperationTestBase transposeAndSqueezeTest(operationCode, {input, dims}, {output}, in transposeAndSqueezeOpTest()
3600 void normalizationOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in normalizationOpTest() argument
3602 OperationTestBase normalizationTest(operationCode, {getOpType(operandCode, 4, inputDim)}, in normalizationOpTest()
3607 operationCode, {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32)}, in normalizationOpTest()
3718 void logicalTest(ANeuralNetworksOperationType operationCode) { in logicalTest() argument
3728 OperationTestBase test(operationCode, {input1, input2}, {output}); in logicalTest()
3740 void comparisonTest(ANeuralNetworksOperationType operationCode, int32_t inputOperandType) { in comparisonTest() argument
3749 OperationTestBase test(operationCode, {input1, input2}, {output}); in comparisonTest()
3807 void reduceOpTest(ANeuralNetworksOperationType operationCode, int32_t inputOperandType) { in reduceOpTest() argument
3814 OperationTestBase test(operationCode, {input1, input2, input3}, {output}, in reduceOpTest()
3851 void selectTest(ANeuralNetworksOperationType operationCode, int32_t inputOperandType) { in selectTest() argument
3858 OperationTestBase test(operationCode, {input0, input1, input2}, {output}); in selectTest()