Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched full:get_tpu_context (Results 1 - 4 of 4) sorted by null

/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_function.py	58 def get_tpu_context(): function
tpu_optimizer.py	76 num_shards = tpu_function.get_tpu_context().number_of_shards
training_loop.py	110 number_of_shards = tpu_function.get_tpu_context().number_of_shards
/external/tensorflow/tensorflow/contrib/kfac/python/ops/
utils.py	329 return tpu_function.get_tpu_context().number_of_shards is not None 346 num_shards = tpu_function.get_tpu_context().number_of_shards

Completed in 641 milliseconds