Searched refs:numElementsPerBatch (Results 1 – 1 of 1) sorted by relevance
65 const unsigned int numElementsPerBatch = inputShape.GetNumElements() / batches; in DepthToSpace() local69 const uintptr_t batchDataOffset = batchIndex * (numElementsPerBatch * dataTypeSize); in DepthToSpace()