tpu_shard_context (reference) in projects: external

Searched refs:tpu_shard_context (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
D	tpu_function.py	`46 def tpu_shard_context(number_of_shards): function`
D	tpu.py	`838 with tpu_function.tpu_shard_context(`