Searched refs:sizeShape (Results 1 – 1 of 1) sorted by relevance
54 const Shape& beginShape, const int32_t* sizeData, const Shape& sizeShape, in evalGeneric() argument119 const Shape& sizeShape = context->getInputShape(kSizeTensor); in prepare() local120 NN_RET_CHECK_EQ(getNumberOfDimensions(sizeShape), 1); in prepare()121 NN_RET_CHECK_EQ(getSizeOfDimension(sizeShape, 0), n_dims); in prepare()