Lines Matching refs:axis
38 inline bool eval(const T* inputData, const Shape& inputShape, int32_t numGroups, int32_t axis, in eval() argument
40 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in eval()
41 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in eval()
43 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
85 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare() local
86 NN_RET_CHECK(handleNegativeAxis(input, &axis)); in prepare()
88 NN_RET_CHECK(getSizeOfDimension(input, axis) % numGroups == 0); in prepare()
94 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute() local
95 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
99 context->getInputShape(kInputTensor), numGroups, axis, in execute()
103 context->getInputShape(kInputTensor), numGroups, axis, in execute()
107 context->getInputShape(kInputTensor), numGroups, axis, in execute()
111 context->getInputShape(kInputTensor), numGroups, axis, in execute()