Home
last modified time | relevance | path

Searched refs:quantize_output (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dtoco_tooling.cc249 const bool quantize_output = in TransformWithStatus() local
253 if (quantize_output) { in TransformWithStatus()
260 UseArraysExtraInfo(model, quantize_output); in TransformWithStatus()
275 if (quantize_output) { in TransformWithStatus()
292 if ((quantize_output && toco_flags.reorder_across_fake_quant())) { in TransformWithStatus()
319 if (quantize_output) { in TransformWithStatus()
351 if (quantize_output) { in TransformWithStatus()
Dtooling_util.h322 void UseArraysExtraInfo(Model* model, bool quantize_output);
Dtooling_util.cc2372 void UseArraysExtraInfo(Model* model, bool quantize_output) { in UseArraysExtraInfo() argument
2389 if (entry.has_data_type() && quantize_output) { in UseArraysExtraInfo()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dpost_quantize.cc74 auto quantize_output = quantize_op.output(); in RemoveQuantizationAdaptorOps() local
75 auto quantize_type = quantize_output.getType(); in RemoveQuantizationAdaptorOps()
78 quantize_output.replaceAllUsesWith(new_arg); in RemoveQuantizationAdaptorOps()