tpu_shard_context (reference) in projects: external

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

/external/tensorflow/tensorflow/python/tpu/
D	tpu_function.py	`47 def tpu_shard_context(number_of_shards): function`
D	feature_column_v2_test.py	`249 with tpu_function.tpu_shard_context(1): 276 with tpu_function.tpu_shard_context(1): 607 with tpu_function.tpu_shard_context(1): 694 with tpu_function.tpu_shard_context(1):`
D	tpu.py	`1403 with tpu_function.tpu_shard_context(`