Searched refs:RerouteTensor (Results 1 – 4 of 4) sorted by relevance
135 nodes_modified_count = common.RerouteTensor(bias_add_tensor,416 common.RerouteTensor(427 common.RerouteTensor(535 nodes_modified_count = common.RerouteTensor(560 nodes_modified_count = common.RerouteTensor(
136 def RerouteTensor(t0, t1, can_modify=None): function
87 common.RerouteTensor(b, a, can_modify=[add_ac.op])
768 tensors_modified_count = common.RerouteTensor(