Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:tpu_function (Results 1 - 8 of 8) sorted by null

/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_function.py	22 from tensorflow.python.tpu.tpu_function import *
keras_support.py	61 from tensorflow.contrib.tpu.python.tpu import tpu_function 350 num_shards = tpu_function.get_tpu_context().number_of_shards     [all...]
/external/tensorflow/tensorflow/python/tpu/
training_loop.py	26 from tensorflow.python.tpu import tpu_function 111 number_of_shards = tpu_function.get_tpu_context().number_of_shards 164 num_replicas = tpu_function.get_tpu_context().number_of_shards
tpu_optimizer.py	26 from tensorflow.python.tpu import tpu_function 123 num_shards = tpu_function.get_tpu_context().number_of_shards
feature_column.py	28 from tensorflow.python.tpu import tpu_function 435 return tpu_function.get_tpu_context().number_of_shards is None
tpu.py	38 from tensorflow.python.tpu import tpu_function     [all...]
tpu_estimator.py	72 from tensorflow.python.tpu import tpu_function     [all...]
/external/tensorflow/tensorflow/python/tpu/ops/
tpu_ops.py	30 from tensorflow.python.tpu import tpu_function 34 num_shards = tpu_function.get_tpu_context().number_of_shards

Completed in 262 milliseconds