Lines Matching refs:OperationTestBase
330 class OperationTestBase { class
332 OperationTestBase(ANeuralNetworksOperationType opCode, in OperationTestBase() function in __anon37b312a30111::OperationTestBase
1044 OperationTestBase test(operationCode, {input0, axis}, {output}); in argMinMaxTest()
1070 OperationTestBase dequantizeTest(ANEURALNETWORKS_DEQUANTIZE, {input}, {output}, in dequantizeOpTest()
1099 OperationTestBase test(ANEURALNETWORKS_EXPAND_DIMS, {input0, axis}, {output}); in expandDimsTest()
1127 OperationTestBase test(ANEURALNETWORKS_GATHER, {input0, axis, input2}, {output}); in gatherTest()
1148 OperationTestBase test(ANEURALNETWORKS_QUANTIZE, {input}, {output}); in quantizeOpTest()
1207 OperationTestBase test( in TEST()
1234 OperationTestBase test(ANEURALNETWORKS_SPLIT, {input0, axis, count}, {output0, output1}); in splitTest()
1258 OperationTestBase test(ANEURALNETWORKS_TILE, {input0, multiples}, {output0}); in tileTest()
1279 OperationTestBase test(ANEURALNETWORKS_TOPK_V2, {input, k}, {outputValues, outputIndices}); in topkV2Test()
1303 OperationTestBase simpleMathTest( in simpleMathOpTest()
1396 OperationTestBase mulTest(ANEURALNETWORKS_MUL, {input1, input2, activation}, {output}); in TEST()
1407 OperationTestBase test(operationCode, {input1, input2}, {output}); in binaryOpTest()
1462 OperationTestBase test(operationCode, {input}, {output}, inputRankMutators); in activationOpTest()
1646 OperationTestBase test(ANEURALNETWORKS_ELU, {input, alpha}, {output}); in eluOpTest()
1666 OperationTestBase test(ANEURALNETWORKS_RESHAPE, {input, shape}, {output}, in reshapeOpTest()
1704 OperationTestBase test(ANEURALNETWORKS_LOG_SOFTMAX, {input, beta, axis}, {output}); in logSoftmaxOpTest()
1723 OperationTestBase test(ANEURALNETWORKS_MEAN, {input, dims, keepDims}, {output}, in meanOpTest()
1753 OperationTestBase test(ANEURALNETWORKS_PAD, {input, padSize}, {output}, in padOpTest()
1781 OperationTestBase test(ANEURALNETWORKS_PAD_V2, {input, padSize, padValue}, {output}, in padV2OpTest()
1803 OperationTestBase softmaxTest(ANEURALNETWORKS_SOFTMAX, {input, beta}, {output}, in softmaxOpTest()
1808 OperationTestBase softmaxAxisTest(ANEURALNETWORKS_SOFTMAX, {input, beta, axis}, {output}, in softmaxOpTest()
1849 OperationTestBase explicitPoolingTest(operationCode, in poolingOpTest()
1856 OperationTestBase implicitPoolingTest( in poolingOpTest()
1868 OperationTestBase explicitNchwPoolingTest( in poolingOpTest()
1875 OperationTestBase implicitNchwPoolingTest(operationCode, in poolingOpTest()
1933 OperationTestBase spaceDepthTest(operationCode, {input, block_size}, {output}); in spaceDepthOpTest()
1941 OperationTestBase spaceDepthNchwTest(operationCode, {input, block_size, layout}, {output}); in spaceDepthOpTest()
1996 OperationTestBase spaceBatchTest(operationCode, {input, blockShape, padding}, {output}); in spaceBatchOpTest()
1999 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, padding, layout}, in spaceBatchOpTest()
2003 OperationTestBase spaceBatchTest(operationCode, {input, blockShape}, {output}); in spaceBatchOpTest()
2006 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, layout}, {output}); in spaceBatchOpTest()
2055 OperationTestBase transposeAndSqueezeTest(operationCode, {input, dims}, {output}, in transposeAndSqueezeOpTest()
2140 OperationTestBase explicitConvTest(ANEURALNETWORKS_CONV_2D, in convOpTest()
2150 OperationTestBase implicitConvTest( in convOpTest()
2165 OperationTestBase explicitNchwConvTest( in convOpTest()
2175 OperationTestBase implicitNchwConvTest( in convOpTest()
2184 OperationTestBase explicitDilateConvTest( in convOpTest()
2194 OperationTestBase implicitDilateConvTest( in convOpTest()
2274 OperationTestBase explicitDepthwiseConvTest( in depthwiseConvOpTest()
2285 OperationTestBase implicitDepthwiseConvTest( in depthwiseConvOpTest()
2300 OperationTestBase explicitNchwDepthwiseConvTest( in depthwiseConvOpTest()
2311 OperationTestBase implicitNchwDepthwiseConvTest(ANEURALNETWORKS_DEPTHWISE_CONV_2D, in depthwiseConvOpTest()
2324 OperationTestBase explicitDilationDepthwiseConvTest( in depthwiseConvOpTest()
2335 OperationTestBase implicitDilationDepthwiseConvTest( in depthwiseConvOpTest()
2399 OperationTestBase fullyConnectedTest(ANEURALNETWORKS_FULLY_CONNECTED, in fullyConnectedOpTest()
2435 OperationTestBase concat2Test(ANEURALNETWORKS_CONCATENATION, {input1, input2, activation}, in concatenationTest()
2439 OperationTestBase concat1Test(ANEURALNETWORKS_CONCATENATION, {input1, activation}, {output}, in concatenationTest()
2468 OperationTestBase resizeTest(ANEURALNETWORKS_RESIZE_BILINEAR, {input, height, width}, {output}); in resizeBilinearOpTest()
2472 OperationTestBase resizeNchwTest(ANEURALNETWORKS_RESIZE_BILINEAR, in resizeBilinearOpTest()
2500 OperationTestBase embedLookupTest(ANEURALNETWORKS_EMBEDDING_LOOKUP, {lookup, input}, {output}); in embeddingLookupTest()
2537 OperationTestBase hashLookupTest(ANEURALNETWORKS_HASHTABLE_LOOKUP, {lookup, keys, values}, in hashtableLookupTest()
2571 OperationTestBase lshProjTest(ANEURALNETWORKS_LSH_PROJECTION, {hash, input, weight, type}, in lshProjectionTest()
2649 OperationTestBase lstmTest(ANEURALNETWORKS_LSTM, in TEST()
2737 OperationTestBase lstmTest(ANEURALNETWORKS_LSTM, in lstmTestV1_2()
2885 OperationTestBase lstmTest(ANEURALNETWORKS_BIDIRECTIONAL_SEQUENCE_LSTM, in lstmBidirectionalSequence()
2987 OperationTestBase multinomialTest(ANEURALNETWORKS_RANDOM_MULTINOMIAL, in randomMultinomialOpTest()
3029 OperationTestBase rnnTest(ANEURALNETWORKS_RNN, in TEST()
3064 OperationTestBase rnnTest(ANEURALNETWORKS_RNN, in TEST()
3100 OperationTestBase svdfTest( in TEST()
3137 OperationTestBase svdfTest( in TEST()
3167 OperationTestBase stridedSliceTest( in stridedSliceOpTest()
3193 OperationTestBase roiAlignTest( in roiAlignOpTest()
3228 OperationTestBase roiPoolingTest( in roiPoolingOpTest()
3262 OperationTestBase heatmapMaxKeypointTest( in heatmapMaxKeypointOpTest()
3303 OperationTestBase test(ANEURALNETWORKS_INSTANCE_NORMALIZATION, in instanceNormalizationOpTest()
3343 OperationTestBase explicitGroupedConvTest(ANEURALNETWORKS_GROUPED_CONV_2D, in groupedConvOpTest()
3352 OperationTestBase implicitGroupedConvTest( in groupedConvOpTest()
3416 OperationTestBase explicitTransposeConvTest( in transposeConvOpTest()
3425 OperationTestBase implicitTransposeConvTest( in transposeConvOpTest()
3465 OperationTestBase channelShuffleTest( in channelShuffleOpTest()
3524 OperationTestBase test(ANEURALNETWORKS_DETECTION_POSTPROCESSING, in detectionPostprocessingOpTest()
3539 OperationTestBase preluTest( in preluOpTest()
3564 OperationTestBase normalizationTest(operationCode, {getOpType(operandCode, 4, inputDim)}, in normalizationOpTest()
3568 OperationTestBase normalizationAxisTest( in normalizationOpTest()
3596 OperationTestBase lrnTest( in localResponseNormOpTest()
3603 OperationTestBase lrnAxisTest( in localResponseNormOpTest()
3623 OperationTestBase axisAlignedBBoxTransformTest( in axisAlignedBBoxTransformOpTest()
3656 OperationTestBase sliceTest(ANEURALNETWORKS_SLICE, in sliceTest()
3690 OperationTestBase test(operationCode, {input1, input2}, {output}); in logicalTest()
3711 OperationTestBase test(operationCode, {input1, input2}, {output}); in comparisonTest()
3776 OperationTestBase test(operationCode, {input1, input2, input3}, {output}, in reduceOpTest()
3820 OperationTestBase test(operationCode, {input0, input1, input2}, {output}); in selectTest()
3840 OperationTestBase test(ANEURALNETWORKS_POW, {inputType, inputType}, {inputType}); in powTest()
3853 OperationTestBase boxWithNmsLimitTest( in boxWithNmsLimitOpTest()
3892 OperationTestBase test(ANEURALNETWORKS_CAST, {input}, {output}); in castOpTest()
3985 OperationTestBase rnnTest(ANEURALNETWORKS_BIDIRECTIONAL_SEQUENCE_RNN, in bidirectionlSequenceRNNTest()
4052 OperationTestBase rnnTest( in unidirectionlSequenceRNNTest()
4181 OperationTestBase ulstmTest(ANEURALNETWORKS_UNIDIRECTIONAL_SEQUENCE_LSTM, in unidirectionalSequenceLSTMTest()
4228 OperationTestBase generateProposalsTest( in generateProposalsOpTest()
4269 OperationTestBase resizeImageOpTest(ANEURALNETWORKS_RESIZE_NEAREST_NEIGHBOR, in resizeNearestNeighborTest()
4379 OperationTestBase test(ANEURALNETWORKS_QUANTIZED_LSTM, in TEST()
4422 OperationTestBase test(ANEURALNETWORKS_FILL, {input0, input1}, {output}); in fillTest()
4442 OperationTestBase test(ANEURALNETWORKS_RANK, {input}, {output}); in rankTest()
4493 OperationTestBase test(ANEURALNETWORKS_IF, {boolType, modelType, modelType, dataType}, in testIf()
4598 OperationTestBase test(ANEURALNETWORKS_WHILE, {modelType, modelType, dataType, dataType}, in testWhile()