Searched refs:outputs_from_all_shards (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 1089 outputs_from_all_shards=True, argument
1212 if isinstance(outputs_from_all_shards, bool):
1213 outputs_from_all_shards = [outputs_from_all_shards] * num_outputs
1215 if num_outputs != len(outputs_from_all_shards):
1220 for (axis, all_shards, x) in zip(output_shard_axes, outputs_from_all_shards,
1239 outputs_from_all_shards=True, argument
1308 outputs_from_all_shards=outputs_from_all_shards,
|
| D | tpu_estimator.py | 3184 outputs_from_all_shards=False,
3209 outputs_from_all_shards=False,
3239 outputs_from_all_shards=False,
|