Home
last modified time | relevance | path

Searched refs:ops_lib (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dsaver_test.py48 from tensorflow.python.framework import ops as ops_lib unknown
83 with self.session(graph=ops_lib.Graph()) as sess:
114 with self.session(graph=ops_lib.Graph()) as sess:
136 with self.session(graph=ops_lib.Graph()) as sess:
171 with ops_lib.Graph().as_default():
173 with ops_lib.device("/job:ps/device:GPU:0"):
178 for op in ops_lib.get_default_graph().get_operations():
192 with ops_lib.Graph().as_default() as g:
222 with self.session(graph=ops_lib.Graph()) as sess:
231 ops_lib._default_graph_stack.reset() # pylint: disable=protected-access
[all …]
Dsaver_test_utils.py23 from tensorflow.python.framework import ops as ops_lib unknown
45 ops_lib.add_to_collection(ops_lib.GraphKeys.SAVEABLE_OBJECTS,
Dcheckpoint_management_test.py31 from tensorflow.python.framework import ops as ops_lib unknown
279 with self.session(graph=ops_lib.Graph()) as sess:
298 with self.session(graph=ops_lib.Graph()) as sess:
313 with self.session(graph=ops_lib.Graph()) as sess:
/external/tensorflow/tensorflow/python/ops/
Dop_selector_test.py22 from tensorflow.python.framework import ops as ops_lib unknown
31 self.graph = ops_lib.Graph()
34 with ops_lib.name_scope("foo"):
38 with ops_lib.name_scope("bar"):
42 with ops_lib.control_dependencies([self.c.op]):
52 g0 = ops_lib.Graph()
56 g1 = ops_lib.Graph()
73 outer = ops_lib.Graph()
86 g0 = ops_lib.Graph()
97 g0 = ops_lib.Graph()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_test.py33 from tensorflow.python.framework import ops as ops_lib unknown
336 with self.session(graph=ops_lib.Graph()) as sess:
348 with self.session(graph=ops_lib.Graph()) as sess:
376 trainable_variables = ops_lib.get_collection(
377 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
395 trainable_variables = ops_lib.get_collection(
396 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
421 with session.Session(config=config, graph=ops_lib.Graph()):
429 with session.Session(config=config, graph=ops_lib.Graph()):
469 with session.Session(config=config, graph=ops_lib.Graph()) as sess:
[all …]
Dsparse_xent_op_test.py33 from tensorflow.python.framework import ops as ops_lib unknown
160 with ops_lib.Graph().as_default(), self.session():
293 with ops_lib.Graph().as_default(), self.session(use_gpu=False) as sess:
308 with ops_lib.device("/cpu:0"): # Sparse-to-dense must be on CPU
360 with ops_lib.device("/cpu:0"):
/external/tensorflow/tensorflow/python/keras/tests/
Dsaver_test.py27 from tensorflow.python.framework import ops as ops_lib unknown
108 save_graph = ops_lib.Graph()
121 restore_graph = ops_lib.Graph()
142 save_graph = ops_lib.Graph()
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_aot_compile.py34 from tensorflow.python.framework import ops as ops_lib unknown
319 with session.Session(graph=ops_lib.Graph()) as sess:
425 temp_graph = ops_lib.Graph()
Dsaved_model_cli.py41 from tensorflow.python.framework import ops as ops_lib unknown
186 with ops_lib.Graph().as_default():
438 with session.Session(worker, graph=ops_lib.Graph()) as sess:
/external/tflite-support/tensorflow_lite_support/custom_ops/
DBUILD12 additional_deps = ["@org_tensorflow_text//tensorflow_text:ops_lib"],