Searched refs:assign_tuple_sharding (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| D | xla_sharding.py | 166 assign_tuple_sharding=False, argument
180 if assign_tuple_sharding:
186 elif assign_tuple_sharding or len(tensor.op.outputs) > 1:
261 def replicate(tensor, assign_tuple_sharding=False, use_sharding_op=False): argument
264 assign_tuple_sharding=assign_tuple_sharding,
270 assign_tuple_sharding=False, argument
275 assign_tuple_sharding=assign_tuple_sharding,
281 assign_tuple_sharding=False, argument
294 assign_tuple_sharding=assign_tuple_sharding,
301 assign_tuple_sharding=False, argument
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_feed.py | 99 return xla_sharding.replicate(tensor, assign_tuple_sharding=True)
101 return xla_sharding.assign_device(tensor, 0, assign_tuple_sharding=True)
107 assign_tuple_sharding=True)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 1072 xla::XlaScopedShardingAssignment assign_tuple_sharding( in BuildArguments() local
|