Searched refs:quant_dim (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_utils.cc | 82 ArrayRef<double> max, int quant_dim, in GetQuantizedType() argument
95 if (min.size() == 1 && max.size() == 1 && quant_dim == -1) { in GetQuantizedType()
106 if (!shape || shape.getRank() <= quant_dim || in GetQuantizedType()
107 static_cast<int64_t>(min.size()) != shape.getDimSize(quant_dim)) { in GetQuantizedType()
112 builder.getUnknownLoc(), storage_type_width, quant_dim, effective_mins, in GetQuantizedType()
156 Attribute max, int quant_dim, in GetQuantizedTypeAttr() argument
182 builder, input_type, min_value, max_value, quant_dim, num_bits.getInt(), in GetQuantizedTypeAttr()
207 Type target, int quant_dim) { in ResetAxisAndBroadcast() argument
229 if (BroadcastVector<double>(shaped.getDimSize(quant_dim), scales) || in ResetAxisAndBroadcast()
230 BroadcastVector<int64_t>(shaped.getDimSize(quant_dim), zero_points)) { in ResetAxisAndBroadcast()
[all …]
|
| D | quantization_utils.h | 499 Attribute max, int quant_dim,
543 Type GetUniformQuantizedPerAxisTypeForWeight(ElementsAttr attr, int quant_dim,
596 ArrayRef<double> max, int quant_dim,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tensorflow/ |
| D | tf_to_quant.cc | 101 int quant_dim = -1; in matchAndRewrite() local
105 quant_dim = res.getType().template cast<ShapedType>().getRank() - 1; in matchAndRewrite()
114 rewriter, res_type, min_value, max_value, quant_dim, num_bits, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 219 int quant_dim = -1; in matchAndRewrite() local
222 quant_dim = res.getType().template cast<ShapedType>().getRank() - 1; in matchAndRewrite()
231 rewriter, res_type, min_value, max_value, quant_dim, num_bits, in matchAndRewrite()
|