Searched refs:firstInputTensorShape (Results 1 – 1 of 1) sorted by relevance
44 const Shape firstInputTensorShape = context->getInputShape(kInputFirstTensor); in prepare() local45 const uint32_t firstInputTensorRank = getNumberOfDimensions(firstInputTensorShape); in prepare()49 NN_RET_CHECK(SameShape(firstInputTensorShape, in prepare()73 outputShape.dimensions[j] = firstInputTensorShape.dimensions[j]; in prepare()77 outputShape.dimensions[j] = firstInputTensorShape.dimensions[j - 1]; in prepare()