OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:TPU
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/python/saved_model/
tag_constants.py
40
# Tag for the `
tpu
` graph.
41
TPU
= "
tpu
"
42
tf_export("saved_model.tag_constants.
TPU
").export_constant(__name__, "
TPU
")
48
"
TPU
"
saved_model_test.py
227
# - multiple tags (from predefined constants for serving on
TPU
).
230
builder.add_meta_graph([tag_constants.SERVING, tag_constants.
TPU
])
262
# Restore the graph with multiple predefined tags (for serving on
TPU
)
265
loader.load(sess, [tag_constants.SERVING, tag_constants.
TPU
], export_dir)
[
all
...]
/external/tensorflow/tensorflow/contrib/saved_model/python/saved_model/
reader_test.py
91
# - multiple predefined tags for serving on
TPU
.
94
builder.add_meta_graph([tag_constants.SERVING, tag_constants.
TPU
])
107
expected_tags = [["train"], ["serve"], ["serve", "gpu"], ["serve", "
tpu
"],
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_estimator.py
33
from tensorflow.contrib.
tpu
.python.ops import tpu_ops
34
from tensorflow.contrib.
tpu
.python.
tpu
import
tpu
35
from tensorflow.contrib.
tpu
.python.
tpu
import tpu_config
36
from tensorflow.contrib.
tpu
.python.
tpu
import tpu_context
37
from tensorflow.contrib.
tpu
.python.
tpu
import tpu_fee
[
all
...]
Completed in 115 milliseconds