Searched refs:per_axis (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 881 if (auto per_axis = in PostProcessFuncOp() local
884 per_axis.getFlags(), per_axis.getStorageType(), in PostProcessFuncOp()
885 per_axis.getExpressedType(), per_axis.getScales(), in PostProcessFuncOp()
886 per_axis.getZeroPoints(), per_axis.getQuantizedDimension(), in PostProcessFuncOp()
887 per_axis.getStorageTypeMin() - 1, per_axis.getStorageTypeMax()); in PostProcessFuncOp()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | operator_property.h | 59 bool per_axis = false; member
|
| D | operator_property.cc | 148 tensor_property.per_axis = true; in GetOperatorProperty()
159 tensor_property.per_axis = true; in GetOperatorProperty()
170 tensor_property.per_axis = true; in GetOperatorProperty()
|
| D | quantize_model.cc | 635 if (utils::QuantizeWeight(model, tensor, tensor_property.per_axis, in QuantizeOpInput()
1130 weight_property.per_axis, weight_property.per_axis_index, in QuantizeBiases()
1212 if (input.second.per_axis) { in FillQuantizationParams()
1228 } else if (!utils::HasMinMax(tensor) && !input.second.per_axis && in FillQuantizationParams()
1360 if (weight_property.per_axis) { in EnsureBiasScaleCompatibility()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_utils.cc | 260 if (auto per_axis = in CastQuantizedTypeAttrFromExpressedType() local
263 ResetAxisAndBroadcast(source_type.getShape(), per_axis, target, axis); in CastQuantizedTypeAttrFromExpressedType()
|