Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
H A Dimport_quant_stats_pass.cc144 ElementsAttr axis_stats, in InsertStatsOpAtResult()
168 ElementsAttr axis_stats; in ImportAsStatsOps() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dconvert_custom_aggregation_op_to_quant_stats.cc93 ElementsAttr axis_stats; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_quantize_helper.h212 mlir::ElementsAttr axis_stats; in processRestrictScale() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_import.cc298 mlir::ElementsAttr axis_stats; in ConvertMinMaxToStatsOp() local
H A Dflatbuffer_export.cc1427 Optional<mlir::ElementsAttr> axis_stats = stats_op.getAxisStats(); in GetQuantizationForQuantStatsOpOutput() local