Searched refs:sliceSize (Results 1 – 2 of 2) sorted by relevance
130 int32_t sliceSize = sizeData[i]; in prepare() local131 if (sliceSize == -1) { in prepare()132 sliceSize = getSizeOfDimension(inputShape, i) - sliceBegin; in prepare()135 NN_RET_CHECK_GE(sliceSize, 0); in prepare()136 NN_RET_CHECK_LE(sliceBegin + sliceSize, getSizeOfDimension(inputShape, i)); in prepare()137 outputShape.dimensions[i] = sliceSize; in prepare()
695 const int32_t sliceSize = sizeOfAxisToSplit / numOutputs; in splitPrepare() local700 output->at(i).dimensions[axis] = sliceSize; in splitPrepare()