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