Searched refs:axisDimension (Results 1 – 2 of 2) sorted by relevance
56 const int32_t axisDimension = context->getInputValue<int32_t>(kInputAxisScalar); in prepare() local57 NN_RET_CHECK_GE(axisDimension, 0); in prepare()58 NN_RET_CHECK_LT(uint32_t(axisDimension), firstInputTensorRank + 1); in prepare()72 for (int32_t j = 0; j < axisDimension; ++j) { in prepare()75 outputShape.dimensions[axisDimension] = context->getNumInputs() - 1; in prepare()76 for (int32_t j = axisDimension + 1; j < int32_t(outputTensorRank); ++j) { in prepare()
52 const int32_t axisDimension = (context->getInputBuffer<int32_t>(kInputAxisTensor))[0]; in prepare() local53 NN_RET_CHECK_GE(axisDimension, 0); in prepare()54 NN_RET_CHECK_LT(uint32_t(axisDimension), inputTensorRank); in prepare()