Lines Matching refs:inputDims
3001 uint32_t inputDims[2] = {5, 5}; in randomMultinomialOpTest() local
3004 .dimensions = inputDims, in randomMultinomialOpTest()
3329 uint32_t inputDims[4] = {4, 4, 4, 4}; in instanceNormalizationOpTest() local
3330 ANeuralNetworksOperandType input = getOpType(inputOperandType, 4, inputDims); in instanceNormalizationOpTest()
3534 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest() local
3535 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDims); in detectionPostprocessingOpTest()
3969 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in bidirectionlSequenceRNNTest() local
3978 .dimensions = inputDims, in bidirectionlSequenceRNNTest()
4045 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionlSequenceRNNTest() local
4054 .dimensions = inputDims, in unidirectionlSequenceRNNTest()
4111 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionalSequenceLSTMTest() local
4124 .dimensions = inputDims, in unidirectionalSequenceLSTMTest()