Searched refs:copy_xla_sharding (Results 1 – 4 of 4) sorted by relevance
58 copy_xla_sharding=False): argument111 if copy_xla_sharding:121 copy_xla_sharding=False): argument159 copy_xla_sharding=copy_xla_sharding)168 copy_xla_sharding=copy_xla_sharding)178 copy_xla_sharding=False): argument218 copy_xla_sharding=copy_xla_sharding)227 copy_xla_sharding=copy_xla_sharding)235 copy_xla_sharding=False): argument262 copy_xla_sharding=copy_xla_sharding)[all …]
187 v, v.initialized_value(), name="slot", copy_xla_sharding=True)200 v, name="slot", dtype=dtypes.float64, copy_xla_sharding=True)
463 copy_xla_sharding=True)474 copy_xla_sharding=True)
1157 var, op_name, copy_xla_sharding=True)