Lines Matching refs:inputDims
2963 uint32_t inputDims[2] = {5, 5}; in randomMultinomialOpTest() local
2966 .dimensions = inputDims, in randomMultinomialOpTest()
3291 uint32_t inputDims[4] = {4, 4, 4, 4}; in instanceNormalizationOpTest() local
3292 ANeuralNetworksOperandType input = getOpType(inputOperandType, 4, inputDims); in instanceNormalizationOpTest()
3496 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest() local
3497 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDims); in detectionPostprocessingOpTest()
3931 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in bidirectionlSequenceRNNTest() local
3940 .dimensions = inputDims, in bidirectionlSequenceRNNTest()
4007 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionlSequenceRNNTest() local
4016 .dimensions = inputDims, in unidirectionlSequenceRNNTest()
4073 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionalSequenceLSTMTest() local
4086 .dimensions = inputDims, in unidirectionalSequenceLSTMTest()