TrainingExtensions/tensorflow/src/python/aimet_tensorflow/common/connectedgraph.py
def _link_previous_op_to_branch_op(self, original_op: Op, branch_op: Op,
                                       output_shape: tf.TensorShape):
        """ Link the original op to the new branch op by creating a product """

        product = Product(original_op.name + '_to_' + branch_op.name, output_shape)
        product.producer = original_op
        product.add_consumer(branch_op)
        original_op.output = product
        branch_op.add_input(product)
        self._products[product.name] = product
 Link the original op to the new branch op by creating a product 
Link the original op to the new branch op by creating a product
Link the original op to the new branch op by creating a product
def _link_previous_op_to_branch_op(self, original_op: Op, branch_op: Op,
                                       output_shape: tf.TensorShape):
        product = Product(original_op.name + '_to_' + branch_op.name, output_shape)
        product.producer = original_op
        product.add_consumer(branch_op)
        original_op.output = product
        branch_op.add_input(product)
        self._products[product.name] = product
def _link_previous_op_to_branch_op ( self , original_op : Op , branch_op : Op , output_shape : tf . TensorShape ) : product = Product ( original_op . name + '_to_' + branch_op . name , output_shape ) product . producer = original_op product . add_consumer ( branch_op ) original_op . output = product branch_op . add_input ( product ) self . _products [ product . name ] = product
Link the original op to the new branch op by creating a product
Link the original op to the new branch op by creating a product
""" Link the original op to the new branch op by creating a product """
self
original_op
Op
branch_op
Op
output_shape
tf.TensorShape