HomeSort by relevance Sort by last modified time
    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