Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DStridedSlice.cpp104 const Shape& endShape = context->getInputShape(kEndTensor); in prepare() local
108 NN_OPS_CHECK(getNumberOfDimensions(endShape) == 1); in prepare()
112 NN_OPS_CHECK(getSizeOfDimension(endShape, 0) == numInputDims); in prepare()
116 NN_OPS_CHECK(endShape.type == OperandType::TENSOR_INT32); in prepare()