Lines Matching refs:inputOperandCode
1138 void quantizeOpTest(int32_t inputOperandCode, int32_t outputOperandCode) { in quantizeOpTest() argument
1141 .type = inputOperandCode, .dimensionCount = 4, .dimensions = inputDimensions}; in quantizeOpTest()
1657 void reshapeOpTest(int32_t inputOperandCode) { in reshapeOpTest() argument
1658 SCOPED_TRACE(inputOperandCode); in reshapeOpTest()
1660 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDimensions); in reshapeOpTest()
1664 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 2, outputDimensions); in reshapeOpTest()
1677 void logSoftmaxOpTest(int32_t inputOperandCode) { in logSoftmaxOpTest() argument
1679 ANeuralNetworksOperandType input = {.type = inputOperandCode, in logSoftmaxOpTest()
1684 ANeuralNetworksOperandType beta = {.type = (inputOperandCode == ANEURALNETWORKS_TENSOR_FLOAT32) in logSoftmaxOpTest()
1697 ANeuralNetworksOperandType output = {.type = inputOperandCode, in logSoftmaxOpTest()
1715 void meanOpTest(int32_t inputOperandCode) { in meanOpTest() argument
1717 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDimensions); in meanOpTest()
1720 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 3, inputDimensions); in meanOpTest()
1743 void padOpTest(int32_t inputOperandCode) { in padOpTest() argument
1744 SCOPED_TRACE(inputOperandCode); in padOpTest()
1746 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 4, inputDimensions); in padOpTest()
1751 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outputDimensions); in padOpTest()
1764 void padV2OpTest(int32_t inputOperandCode) { in padV2OpTest() argument
1765 SCOPED_TRACE(inputOperandCode); in padV2OpTest()
1767 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 4, inputDimensions); in padV2OpTest()
1772 if (inputOperandCode == ANEURALNETWORKS_TENSOR_FLOAT16) { in padV2OpTest()
1774 } else if (inputOperandCode == ANEURALNETWORKS_TENSOR_QUANT8_ASYMM || in padV2OpTest()
1775 inputOperandCode == ANEURALNETWORKS_TENSOR_QUANT8_ASYMM_SIGNED) { in padV2OpTest()
1779 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outputDimensions); in padV2OpTest()
2092 void convOpTest(int32_t inputOperandCode, int32_t filterOperandCode) { in convOpTest() argument
2094 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 4, inputDimensions); in convOpTest()
2106 ANeuralNetworksOperandType bias = {.type = inputOperandCode, in convOpTest()
2230 void depthwiseConvOpTest(int32_t inputOperandCode, int32_t filterOperandCode) { in depthwiseConvOpTest() argument
2232 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 4, inputDimensions); in depthwiseConvOpTest()
2244 ANeuralNetworksOperandType bias = {.type = inputOperandCode, in depthwiseConvOpTest()
2459 void resizeBilinearOpTest(int32_t inputOperandCode, int32_t scalarOperandCode) { in resizeBilinearOpTest() argument
2460 SCOPED_TRACE(inputOperandCode); in resizeBilinearOpTest()
2462 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 4, inputDimensions); in resizeBilinearOpTest()
3189 void roiAlignOpTest(int32_t inputOperandCode, int32_t roiOperandCode, int32_t scalarOperandCode) { in roiAlignOpTest() argument
3194 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiAlignOpTest()
3200 {getOpType(inputOperandCode, 4, outDim)}); in roiAlignOpTest()
3224 void roiPoolingOpTest(int32_t inputOperandCode, int32_t roiOperandCode, int32_t scalarOperandCode) { in roiPoolingOpTest() argument
3229 {getOpType(inputOperandCode, 4, inDim), getOpType(roiOperandCode, 2, roiDim), in roiPoolingOpTest()
3234 {getOpType(inputOperandCode, 4, outDim)}); in roiPoolingOpTest()
3312 void groupedConvOpTest(int32_t inputOperandCode, int32_t filterOperandCode) { in groupedConvOpTest() argument
3315 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 4, inDim); in groupedConvOpTest()
3326 ANeuralNetworksOperandType bias = getOpType(inputOperandCode, 1, biasDim); in groupedConvOpTest()
3340 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outDim); in groupedConvOpTest()
3387 void transposeConvOpTest(int32_t inputOperandCode, int32_t filterOperandCode) { in transposeConvOpTest() argument
3390 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 4, inDim); in transposeConvOpTest()
3400 ANeuralNetworksOperandType bias = getOpType(inputOperandCode, 1, biasDim); in transposeConvOpTest()
3413 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 4, outDim); in transposeConvOpTest()
3488 void detectionPostprocessingOpTest(int32_t inputOperandCode) { in detectionPostprocessingOpTest() argument
3489 SCOPED_TRACE(inputOperandCode); in detectionPostprocessingOpTest()
3496 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDims); in detectionPostprocessingOpTest()
3498 ANeuralNetworksOperandType deltas = getOpType(inputOperandCode, 3, deltasDims); in detectionPostprocessingOpTest()
3500 ANeuralNetworksOperandType anchors = getOpType(inputOperandCode, 2, anchorsDims); in detectionPostprocessingOpTest()
3502 if (inputOperandCode == ANEURALNETWORKS_TENSOR_FLOAT16) { in detectionPostprocessingOpTest()
3515 ANeuralNetworksOperandType outputScore = getOpType(inputOperandCode, 2, outputScoreDims); in detectionPostprocessingOpTest()
3517 ANeuralNetworksOperandType boundingBoxes = getOpType(inputOperandCode, 3, boundingBoxesDims); in detectionPostprocessingOpTest()
3885 void castOpTest(int32_t inputOperandCode, int32_t outputOperandCode) { in castOpTest() argument
3887 << "inputType: " << inputOperandCode << ", outputType: " << outputOperandCode); in castOpTest()
3889 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDimensions); in castOpTest()
3924 void bidirectionlSequenceRNNTest(int32_t inputOperandCode) { in bidirectionlSequenceRNNTest() argument
3937 ANeuralNetworksOperandType input = {.type = inputOperandCode, in bidirectionlSequenceRNNTest()
3942 ANeuralNetworksOperandType fwWeights = {.type = inputOperandCode, in bidirectionlSequenceRNNTest()
3948 ANeuralNetworksOperandType fwRecurrentWeights = {.type = inputOperandCode, in bidirectionlSequenceRNNTest()
3954 ANeuralNetworksOperandType fwBias = {.type = inputOperandCode, in bidirectionlSequenceRNNTest()
3960 ANeuralNetworksOperandType fwHiddenState = {.type = inputOperandCode, in bidirectionlSequenceRNNTest()
3966 ANeuralNetworksOperandType output = {.type = inputOperandCode, in bidirectionlSequenceRNNTest()
4000 void unidirectionlSequenceRNNTest(int32_t inputOperandCode) { in unidirectionlSequenceRNNTest() argument
4013 ANeuralNetworksOperandType input = {.type = inputOperandCode, in unidirectionlSequenceRNNTest()
4018 ANeuralNetworksOperandType weights = {.type = inputOperandCode, in unidirectionlSequenceRNNTest()
4023 ANeuralNetworksOperandType recurrentWeights = {.type = inputOperandCode, in unidirectionlSequenceRNNTest()
4028 ANeuralNetworksOperandType bias = {.type = inputOperandCode, in unidirectionlSequenceRNNTest()
4033 ANeuralNetworksOperandType hiddenState = {.type = inputOperandCode, in unidirectionlSequenceRNNTest()
4038 ANeuralNetworksOperandType output = {.type = inputOperandCode, in unidirectionlSequenceRNNTest()
4065 void unidirectionalSequenceLSTMTest(int32_t inputOperandCode) { in unidirectionalSequenceLSTMTest() argument
4083 ANeuralNetworksOperandType input = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4088 ANeuralNetworksOperandType inputToInputWeights = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4096 ANeuralNetworksOperandType recurrentToInputWeights = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4104 ANeuralNetworksOperandType cellToInputWeights = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4111 ANeuralNetworksOperandType inputGateBias = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4119 ANeuralNetworksOperandType projectionWeights = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4124 ANeuralNetworksOperandType projectionBias = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4129 ANeuralNetworksOperandType outputStateIn = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4134 ANeuralNetworksOperandType cellStateIn = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4148 .type = inputOperandCode == ANEURALNETWORKS_TENSOR_FLOAT32 ? ANEURALNETWORKS_FLOAT32 in unidirectionalSequenceLSTMTest()
4165 ANeuralNetworksOperandType inputLayerNormWeights = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()
4174 ANeuralNetworksOperandType output = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest()