Lines Matching refs:axis
40 inline bool eval(const T* inputData, const Shape& inputShape, int32_t axis, in eval() argument
42 const auto outerSize = getNumberOfElements(inputShape, 0, axis); in eval()
43 const auto axisSize = getSizeOfDimension(inputShape, axis); in eval()
45 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
84 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare() local
85 NN_RET_CHECK(handleNegativeAxis(input, &axis)); in prepare()
92 input.dimensions.begin() + axis); in prepare()
95 output.dimensions.insert(output.dimensions.end(), input.dimensions.begin() + axis + 1, in prepare()
102 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute() local
103 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
107 context->getInputShape(kInputTensor), axis, in execute()
113 context->getInputShape(kInputTensor), axis, in execute()
119 context->getInputShape(kInputTensor), axis, in execute()
125 context->getInputShape(kInputTensor), axis, in execute()
131 context->getInputShape(kInputTensor), axis, in execute()