13ba9a024d50038d94706beeb5267a88a1a2ac5e,mmdnn/conversion/tensorflow/tensorflow_emitter.py,TensorflowEmitter,emit_Slice,#TensorflowEmitter#Any#,584

Before Change

            extra_str += ", begin_mask={}".format(IR_node.get_attr("begin_mask"))
        if IR_node.get_attr("end_mask"):
            extra_str += ", end_mask={}".format(IR_node.get_attr("end_mask"))
        self.add_body(1, "{:<15} = tf.strided_slice({}, {}, {} {}, name="{}")".format(
            IR_node.variable_name,
            self.parent_variable_name(IR_node),
            IR_node.get_attr("starts"),
            IR_node.get_attr("ends"),
            extra_str,
            IR_node.name))

    def emit_Shape(self, IR_node):
        self.add_body(1, "{:<15} = tf.shape({}, name="{}")".format(
            IR_node.variable_name,

After Change

            extra_str += ", begin_mask={}".format(IR_node.get_attr("begin_mask"))
        if IR_node.get_attr("end_mask"):
            extra_str += ", end_mask={}".format(IR_node.get_attr("end_mask"))
        if IR_node.get_attr("shrink_axis_mask"):
            extra_str += ", shrink_axis_mask={}".format(IR_node.get_attr("shrink_axis_mask"))
        
        code = "{:<15} = tf.strided_slice({}, {}, {} {}, name="{}")".format(
            IR_node.variable_name,
            self.parent_variable_name(IR_node),
            IR_node.get_attr("starts"),

Instances