[luci-pass] Add cal_offset_2d

Add a function for calculating offset of 2d tensor.

ONE-DCO-1.0-Signed-off-by: young cheon <[email protected]>

compiler/luci/pass/src/QuantizationUtils.cpp

@@ -292,6 +292,11 @@ uint32_t cal_offset(loco::TensorShape &dimension, uint32_t *indices)
          indices[2] * dimension.dim(3).value() + indices[3];
 }
 
+uint32_t cal_offset_2d(loco::TensorShape &dimension, uint32_t *indices)
+{
+  return indices[0] * dimension.dim(1).value() + indices[1];
+}
+
 // Activation (ofm) qtype is determined in different ways.
 // 1. Pre-defined values: Some Ops have pre-defined qparams (ex: LOGISTIC, TANH)
 // 2. Integer scale: Output of some Ops should be integers (ex: FLOOR, CEIL)

compiler/luci/pass/src/QuantizationUtils.h

@@ -53,6 +53,7 @@ bool get_channel_dim_index(CircleConst *node, loco::TensorShape &dimension,
 
 // Calculate offset of the given indices in dimension
 uint32_t cal_offset(loco::TensorShape &dimension, uint32_t *indices);
+uint32_t cal_offset_2d(loco::TensorShape &dimension, uint32_t *indices);
 
 // Backward propagation of concatenation qparam
 void propagate_concat_quantparam(luci::CircleConcatenation *concat);
@@ -63,6 +64,8 @@ void propagate_pad_v2_quantparam(luci::CirclePadV2 *pad_v2);
 // Return true if the node is quantized
 bool is_quantized(const CircleNode *node);
 
+uint8_t fp32_to_uint8_cast(float f);
+
 // Return true if the node is fp32
 bool is_fp32(const CircleNode *node);