Searched refs:blockSizeDim (Results 1 – 1 of 1) sorted by relevance
190 tflite::Dims<4> blockSizeDim, cropsDim; in batchToSpaceGeneric() local194 blockSizeDim, crops, cropsDim, outputData, in batchToSpaceGeneric()213 tflite::RuntimeShape blockSizeDim; in spaceToBatchGeneric() local217 blockSizeDim, blockSize, convertShapeToTflshape(paddingShape), padding, in spaceToBatchGeneric()