Home
last modified time | relevance | path

Searched refs:min_maxs (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dimport_quant_stats_pass.cc140 SmallVector<APFloat, 4> min_maxs; in ImportAsStatsOps() local
141 min_maxs.reserve(info.params_size() * 2); in ImportAsStatsOps()
145 min_maxs.push_back(min); in ImportAsStatsOps()
146 min_maxs.push_back(max); in ImportAsStatsOps()
150 RankedTensorType::get({2}, b.getF32Type()), {min_maxs[0], min_maxs[1]}); in ImportAsStatsOps()
157 RankedTensorType::get(axis_stats_shape, b.getF32Type()), min_maxs); in ImportAsStatsOps()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc253 llvm::SmallVector<llvm::APFloat, 4> min_maxs; in ConvertMinMaxToStatsOp() local
254 min_maxs.reserve(mins.size() * 2); in ConvertMinMaxToStatsOp()
258 min_maxs.push_back(min); in ConvertMinMaxToStatsOp()
259 min_maxs.push_back(max); in ConvertMinMaxToStatsOp()
264 {min_maxs[0], min_maxs[1]}); in ConvertMinMaxToStatsOp()
272 min_maxs); in ConvertMinMaxToStatsOp()