Home
last modified time | relevance | path

Searched refs:sizeShape (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DSlice.cpp54 const Shape& beginShape, const int32_t* sizeData, const Shape& sizeShape, in evalGeneric() argument
119 const Shape& sizeShape = context->getInputShape(kSizeTensor); in prepare() local
120 NN_RET_CHECK_EQ(getNumberOfDimensions(sizeShape), 1); in prepare()
121 NN_RET_CHECK_EQ(getSizeOfDimension(sizeShape, 0), n_dims); in prepare()