Lines Matching refs:axis
48 inline bool l2normFloat32Impl(const float* inputData, const Shape& inputShape, int32_t axis, in l2normFloat32Impl() argument
52 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normFloat32Impl()
53 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normFloat32Impl()
55 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normFloat32Impl()
76 inline bool l2normQuant8Impl(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Impl() argument
79 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8Impl()
80 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8Impl()
82 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8Impl()
108 inline bool l2normQuant8SignedImpl(const int8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8SignedImpl() argument
111 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8SignedImpl()
112 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8SignedImpl()
114 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8SignedImpl()
139 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32() argument
142 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normFloat32()
144 if (axis == ndim - 1) { in l2normFloat32()
151 return l2normFloat32Impl(inputData, inputShape, axis, outputData, outputShape); in l2normFloat32()
155 bool l2normFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in l2normFloat16() argument
162 l2normFloat32(inputDataFloat32.data(), inputShape, axis, outputDataFloat32.data(), outputShape); in l2normFloat16()
168 bool l2normQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8() argument
171 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8()
173 if (axis == ndim - 1) { in l2normQuant8()
180 return l2normQuant8Impl(inputData, inputShape, axis, outputData, outputShape); in l2normQuant8()
184 bool l2normQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Signed() argument
187 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8Signed()
189 if (axis == ndim - 1) { in l2normQuant8Signed()
191 const int32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8Signed()
192 const int32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8Signed()
197 return l2normQuant8SignedImpl(inputData, inputShape, axis, outputData, outputShape); in l2normQuant8Signed()
240 int32_t axis = context->getNumInputs() == kNumInputs in prepare() local
244 NN_RET_CHECK_GE(axis, -numDimensions); in prepare()
245 NN_RET_CHECK_LT(axis, numDimensions); in prepare()
263 int32_t axis = context->getNumInputs() == kNumInputs in execute() local
266 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
270 context->getInputShape(kInputTensor), axis, in execute()
275 context->getInputShape(kInputTensor), axis, in execute()
280 context->getInputShape(kInputTensor), axis, in execute()
285 context->getInputShape(kInputTensor), axis, in execute()