Searched refs:set_prim_instance_name (Results 1 – 4 of 4) sorted by relevance
41 all_reduce_grad.set_prim_instance_name(instance_name)175 all_reduce.set_prim_instance_name(instance_name)220 reduce_scatter.set_prim_instance_name(instance_name)243 all_reduce.set_prim_instance_name(instance_name)278 all_reduce.set_prim_instance_name(instance_name)300 host_all_gather_grad.set_prim_instance_name(instance_name)315 reduce_scatter_grad.set_prim_instance_name(instance_name)333 all_swap_grad.set_prim_instance_name(instance_name)348 host_reduce_scatter_grad.set_prim_instance_name(instance_name)383 all_to_all_grad.set_prim_instance_name(instance_name)[all …]
92 cloned.set_prim_instance_name(self.instance_name)183 def set_prim_instance_name(self, instance_name): member in Primitive
265 op.set_prim_instance_name(instance_name)
549 value.set_prim_instance_name(name)