Lines Matching refs:operationCode
327 void argMinMaxTest(ANeuralNetworksOperationType operationCode, int32_t inputOperandType) { in argMinMaxTest() argument
342 OperationTestBase test(operationCode, {input0, axis}, {output}); in argMinMaxTest()
578 void simpleMathOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in simpleMathOpTest() argument
590 OperationTestBase simpleMathTest(operationCode, {input1, input2, activation}, {output}); in simpleMathOpTest()
657 void binaryOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in binaryOpTest() argument
664 OperationTestBase test(operationCode, {input1, input2}, {output}); in binaryOpTest()
700 void activationOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in activationOpTest() argument
705 OperationTestBase test(operationCode, {input}, {output}); in activationOpTest()
994 void poolingOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in poolingOpTest() argument
1014 OperationTestBase explicitPoolingTest(operationCode, in poolingOpTest()
1022 operationCode, in poolingOpTest()
1034 operationCode, in poolingOpTest()
1040 OperationTestBase implicitNchwPoolingTest(operationCode, in poolingOpTest()
1079 void spaceDepthOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in spaceDepthOpTest() argument
1090 OperationTestBase spaceDepthTest(operationCode, {input, block_size}, {output}); in spaceDepthOpTest()
1098 OperationTestBase spaceDepthNchwTest(operationCode, {input, block_size, layout}, {output}); in spaceDepthOpTest()
1126 void spaceBatchOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in spaceBatchOpTest() argument
1144 if (operationCode == ANEURALNETWORKS_SPACE_TO_BATCH_ND) { in spaceBatchOpTest()
1145 OperationTestBase spaceBatchTest(operationCode, {input, blockShape, padding}, {output}); in spaceBatchOpTest()
1148 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, padding, layout}, in spaceBatchOpTest()
1152 OperationTestBase spaceBatchTest(operationCode, {input, blockShape}, {output}); in spaceBatchOpTest()
1155 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, layout}, {output}); in spaceBatchOpTest()
1184 void transposeAndSqueezeOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in transposeAndSqueezeOpTest() argument
1196 OperationTestBase transposeAndSqueezeTest(operationCode, {input, dims}, {output}); in transposeAndSqueezeOpTest()
2590 void normalizationOpTest(ANeuralNetworksOperationType operationCode, int32_t operandCode) { in normalizationOpTest() argument
2592 OperationTestBase normalizationTest(operationCode, {getOpType(operandCode, 4, inputDim)}, in normalizationOpTest()
2597 operationCode, {getOpType(operandCode, 4, inputDim), getOpType(ANEURALNETWORKS_INT32)}, in normalizationOpTest()
2695 void logicalTest(ANeuralNetworksOperationType operationCode) { in logicalTest() argument
2705 OperationTestBase test(operationCode, {input1, input2}, {output}); in logicalTest()
2717 void comparisonTest(ANeuralNetworksOperationType operationCode, int32_t inputOperandType) { in comparisonTest() argument
2726 OperationTestBase test(operationCode, {input1, input2}, {output}); in comparisonTest()
2778 void reduceOpTest(ANeuralNetworksOperationType operationCode, int32_t inputOperandType) { in reduceOpTest() argument
2805 OperationTestBase test(operationCode, {input1, input2, input3}, {output}); in reduceOpTest()
2839 void selectTest(ANeuralNetworksOperationType operationCode, int32_t inputOperandType) { in selectTest() argument
2846 OperationTestBase test(operationCode, {input0, input1, input2}, {output}); in selectTest()