Searched refs:firstInputTensorRank (Results 1 – 1 of 1) sorted by relevance
45 const uint32_t firstInputTensorRank = getNumberOfDimensions(firstInputTensorShape); in prepare() local46 NN_RET_CHECK_GE(firstInputTensorRank, 1U); in prepare()58 NN_RET_CHECK_LT(uint32_t(axisDimension), firstInputTensorRank + 1); in prepare()62 const uint32_t outputTensorRank = firstInputTensorRank + 1; in prepare()