Searched refs:beginShape (Results 1 – 2 of 2) sorted by relevance
53 const Shape& beginShape, const int32_t* sizeData, const Shape& sizeShape, in evalGeneric() argument59 std::vector<uint32_t> beginIndex(getSizeOfDimension(beginShape, 0)); in evalGeneric()110 const Shape& beginShape = context->getInputShape(kBeginTensor); in prepare() local111 NN_RET_CHECK_EQ(getNumberOfDimensions(beginShape), 1); in prepare()112 NN_RET_CHECK_EQ(getSizeOfDimension(beginShape, 0), n_dims); in prepare()
144 const Shape& beginShape = context->getInputShape(kBeginTensor); in prepare() local148 NN_OPS_CHECK(getNumberOfDimensions(beginShape) == 1); in prepare()152 NN_OPS_CHECK(getSizeOfDimension(beginShape, 0) == numInputDims); in prepare()156 NN_OPS_CHECK(beginShape.type == OperandType::TENSOR_INT32); in prepare()