Searched refs:axis_stats (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | import_quant_stats_pass.cc | 78 ElementsAttr axis_stats, IntegerAttr axis);
127 ElementsAttr axis_stats, in InsertStatsOpAtResult() argument
130 layer_stats, axis_stats, axis); in InsertStatsOpAtResult()
151 ElementsAttr axis_stats; in ImportAsStatsOps() local
156 axis_stats = DenseFPElementsAttr::get( in ImportAsStatsOps()
163 InsertStatsOpAtResult(b, op->getResult(index), layer_stats, axis_stats, in ImportAsStatsOps()
168 InsertStatsOpAtResult(b, op->getResult(i), layer_stats, axis_stats, in ImportAsStatsOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_quantize_helper.h | 211 mlir::ElementsAttr axis_stats; in processRestrictScale() local
216 stats_op, stats_op.arg(), layer_stats, axis_stats, axis); in processRestrictScale()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 265 mlir::ElementsAttr axis_stats; in ConvertMinMaxToStatsOp() local
270 axis_stats = mlir::DenseFPElementsAttr::get( in ConvertMinMaxToStatsOp()
277 layer_stats, axis_stats, axis); in ConvertMinMaxToStatsOp()
|
| D | flatbuffer_export.cc | 1311 Optional<mlir::ElementsAttr> axis_stats = stats_op.axisStats(); in GetQuantizationForQuantStatsOpOutput() local
1315 axis_stats.hasValue() in GetQuantizationForQuantStatsOpOutput()
1316 ? axis_stats.getValue().cast<mlir::DenseFPElementsAttr>() in GetQuantizationForQuantStatsOpOutput()
|