Lines Matching refs:axis
41 inline bool eval(const T* inputData, const Shape& inputShape, int32_t numGroups, int32_t axis, in eval() argument
43 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in eval()
44 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in eval()
46 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
88 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare() local
89 NN_RET_CHECK(handleNegativeAxis(input, &axis)); in prepare()
91 NN_RET_CHECK(getSizeOfDimension(input, axis) % numGroups == 0); in prepare()
97 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute() local
98 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
102 context->getInputShape(kInputTensor), numGroups, axis, in execute()
106 context->getInputShape(kInputTensor), numGroups, axis, in execute()
110 context->getInputShape(kInputTensor), numGroups, axis, in execute()
114 context->getInputShape(kInputTensor), numGroups, axis, in execute()