Lines Matching refs:axis
51 const std::vector<Shape>& inputShapes, int32_t axis, T* outputData, in concatenation() argument
63 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
71 const std::vector<Shape>& inputShapes, int32_t axis, in concatenation() argument
88 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
187 int32_t axis = context->getInputValue<int32_t>(numInputs - 1); in prepare() local
188 NN_RET_CHECK_GE(axis, 0); in prepare()
189 NN_RET_CHECK_LT(axis, numDimensions); in prepare()
192 uint32_t sumAxis = getSizeOfDimension(input0, axis); in prepare()
198 if (d == axis) { in prepare()
199 sumAxis += getSizeOfDimension(input, axis); in prepare()
209 output.dimensions[axis] = sumAxis; in prepare()