Lines Matching refs:inputShape

39 inline bool l2normFloat32Impl(const float* inputData, const Shape& inputShape, int32_t axis,  in l2normFloat32Impl()  argument
42 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normFloat32Impl()
43 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normFloat32Impl()
45 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normFloat32Impl()
66 inline bool l2normQuant8Impl(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Impl() argument
69 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8Impl()
70 const uint32_t axisSize = getSizeOfDimension(inputShape, axis); in l2normQuant8Impl()
72 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8Impl()
80 int32_t val = static_cast<int32_t>(*p) - inputShape.offset; in l2normQuant8Impl()
87 int32_t val = static_cast<int32_t>(*p) - inputShape.offset; in l2normQuant8Impl()
98 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32() argument
100 int32_t ndim = getNumberOfDimensions(inputShape); in l2normFloat32()
101 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normFloat32()
106 tflite::optimized_ops::L2Normalization(param, convertShapeToTflshape(inputShape), inputData, in l2normFloat32()
110 return l2normFloat32Impl(inputData, inputShape, axis, outputData, outputShape); in l2normFloat32()
114 bool l2normFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in l2normFloat16() argument
117 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in l2normFloat16()
121 l2normFloat32(inputDataFloat32.data(), inputShape, axis, outputDataFloat32.data(), outputShape); in l2normFloat16()
127 bool l2normQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8() argument
129 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8()
130 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8()
134 tflite::L2NormalizationParams param = {.input_zero_point = inputShape.offset}; in l2normQuant8()
135 tflite::optimized_ops::L2Normalization(param, convertShapeToTflshape(inputShape), inputData, in l2normQuant8()
139 return l2normQuant8Impl(inputData, inputShape, axis, outputData, outputShape); in l2normQuant8()