Searched refs:dequantize_node (Results 1 – 2 of 2) sorted by relevance
284 const NodeDef& dequantize_node = match.inputs[0].node; in RemoveRedundantQuantizations() local286 dequantize_node.input(0); in RemoveRedundantQuantizations()288 dequantize_node.input(1); in RemoveRedundantQuantizations()290 dequantize_node.input(2); in RemoveRedundantQuantizations()295 if (output_nodes.count(dequantize_node.name()) || in RemoveRedundantQuantizations()296 graph_outputs.count(dequantize_node.name())) { in RemoveRedundantQuantizations()358 NodeDef dequantize_node; in QuantizePlaceholders() local359 dequantize_node.set_op("Dequantize"); in QuantizePlaceholders()360 dequantize_node.set_name(namespace_prefix + "/dequantize"); in QuantizePlaceholders()361 SetNodeAttr("T", DT_QUINT8, &dequantize_node); in QuantizePlaceholders()[all …]
126 NodeDef dequantize_node; in QuantizeWeights() local127 dequantize_node.set_op("Dequantize"); in QuantizeWeights()128 dequantize_node.set_name(old_const_node.name()); in QuantizeWeights()129 SetNodeAttr("T", DT_QUINT8, &dequantize_node); in QuantizeWeights()130 SetNodeAttr("mode", "MIN_FIRST", &dequantize_node); in QuantizeWeights()131 AddNodeInput(quantized_const_node.name(), &dequantize_node); in QuantizeWeights()132 AddNodeInput(min_node.name(), &dequantize_node); in QuantizeWeights()133 AddNodeInput(max_node.name(), &dequantize_node); in QuantizeWeights()134 new_nodes->push_back(dequantize_node); in QuantizeWeights()