Lines Matching refs:inputDims
2962 uint32_t inputDims[2] = {5, 5}; in randomMultinomialOpTest() local
2965 .dimensions = inputDims, in randomMultinomialOpTest()
3290 uint32_t inputDims[4] = {4, 4, 4, 4}; in instanceNormalizationOpTest() local
3291 ANeuralNetworksOperandType input = getOpType(inputOperandType, 4, inputDims); in instanceNormalizationOpTest()
3495 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest() local
3496 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDims); in detectionPostprocessingOpTest()
3930 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in bidirectionlSequenceRNNTest() local
3939 .dimensions = inputDims, in bidirectionlSequenceRNNTest()
4006 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionlSequenceRNNTest() local
4015 .dimensions = inputDims, in unidirectionlSequenceRNNTest()
4072 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionalSequenceLSTMTest() local
4085 .dimensions = inputDims, in unidirectionalSequenceLSTMTest()