Home
last modified time | relevance | path

Searched defs:axis_stats (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dimport_quant_stats_pass.cc137 ElementsAttr axis_stats, in InsertStatsOpAtResult()
161 ElementsAttr axis_stats; in ImportAsStatsOps() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h211 mlir::ElementsAttr axis_stats; in processRestrictScale() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc285 mlir::ElementsAttr axis_stats; in ConvertMinMaxToStatsOp() local
Dflatbuffer_export.cc1329 Optional<mlir::ElementsAttr> axis_stats = stats_op.axisStats(); in GetQuantizationForQuantStatsOpOutput() local