f22b843a42b1cd65e34ee7e70866145ceb98f15e,server/bert_serving/server/graph.py,,convert_variables_to_constants,#Any#Any#Any#Any#Any#Any#Any#,146

Before Change

            output_node.CopyFrom(input_node)
            dtype = input_node.attr["dtype"]
            if use_fp16 and dtype.type == types_pb2.DT_FLOAT:
                print("< %s | %s " % (input_node.name, input_node.attr["dtype"]))
                output_node.attr["dtype"].CopyFrom(
                    attr_value_pb2.AttrValue(type=types_pb2.DT_HALF))
                print("> %s | %s " % (output_node.name, output_node.attr["dtype"]))

After Change

            dtype = input_node.attr["dtype"]
            if use_fp16 and dtype.type == types_pb2.DT_FLOAT:
                new_dtype = attr_value_pb2.AttrValue()
                new_dtype.CopyFrom(dtype)
                new_dtype.type.CopyFrom(types_pb2.DT_HALF)
                output_node.attr["dtype"].CopyFrom(new_dtype)
        output_graph_def.node.extend([output_node])

Instances