Searched refs:quantize_node (Results 1 – 1 of 1) sorted by relevance
283 const NodeDef& quantize_node = match.node; in RemoveRedundantQuantizations() local285 inputs_to_rename[quantize_node.name() + ":0"] = in RemoveRedundantQuantizations()287 inputs_to_rename[quantize_node.name() + ":1"] = in RemoveRedundantQuantizations()289 inputs_to_rename[quantize_node.name() + ":2"] = in RemoveRedundantQuantizations()417 NodeDef quantize_node; in ConvertFakeQuantsToRequantize() local418 quantize_node.set_op("QuantizeV2"); in ConvertFakeQuantsToRequantize()419 quantize_node.set_name(namespace_prefix + "/quantize"); in ConvertFakeQuantsToRequantize()420 SetNodeAttr("T", DT_QINT32, &quantize_node); in ConvertFakeQuantsToRequantize()421 SetNodeAttr("mode", "MIN_FIRST", &quantize_node); in ConvertFakeQuantsToRequantize()422 AddNodeInput(fake_quant_node.input(0), &quantize_node); in ConvertFakeQuantsToRequantize()[all …]