| /external/tensorflow/tensorflow/python/client/ |
| D | session_list_devices_test.py | 24 from tensorflow.python import pywrap_tensorflow as tf_session unknown
44 opts = tf_session.TF_NewSessionOptions()
45 c_session = tf_session.TF_NewSession(ops.get_default_graph()._c_graph, opts)
46 raw_device_list = tf_session.TF_SessionListDevices(c_session)
47 size = tf_session.TF_DeviceListCount(raw_device_list)
49 tf_session.TF_DeviceListMemoryBytes(raw_device_list, size)
50 tf_session.TF_DeleteDeviceList(raw_device_list)
51 tf_session.TF_CloseSession(c_session)
|
| D | session.py | 30 from tensorflow.python import pywrap_tensorflow as tf_session unknown
673 opts = tf_session.TF_NewSessionOptions(target=self._target, config=config)
676 self._session = tf_session.TF_NewSessionRef(self._graph._c_graph, opts)
679 tf_session.TF_DeleteSessionOptions(opts)
704 raw_device_list = tf_session.TF_SessionListDevices(self._session)
706 size = tf_session.TF_DeviceListCount(raw_device_list)
708 name = tf_session.TF_DeviceListName(raw_device_list, i)
709 device_type = tf_session.TF_DeviceListType(raw_device_list, i)
710 memory = tf_session.TF_DeviceListMemoryBytes(raw_device_list, i)
711 incarnation = tf_session.TF_DeviceListIncarnation(raw_device_list, i)
[all …]
|
| D | tf_session_helper.cc | 48 TF_Session* tf_session = TF_NewSession(graph, opts, status); in TF_NewSessionRef() local
49 if (tf_session == nullptr) { in TF_NewSessionRef()
53 Session* session = reinterpret_cast<Session*>(tf_session->session); in TF_NewSessionRef()
55 tf_session->session = session_ref; in TF_NewSessionRef()
56 return tf_session; in TF_NewSessionRef()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
| D | trainer_hooks_test.py | 26 from tensorflow.python.client import session as tf_session unknown
56 with ops.Graph().as_default(), tf_session.Session() as sess:
|
| D | custom_export_strategy.py | 30 from tensorflow.python.client import session as tf_session unknown
77 with tf_session.Session(graph=graph) as sess:
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | export.py | 30 from tensorflow.python.client import session as tf_session unknown
74 with tf_session.Session('') as session:
|
| /external/tensorflow/tensorflow/examples/ios/camera/ |
| D | CameraExampleViewController.h | 38 std::unique_ptr<tensorflow::Session> tf_session; variable
|
| D | CameraExampleViewController.mm | 315 if (tf_session.get()) {
317 tensorflow::Status run_status = tf_session->Run(
386 model_file_name, model_file_type, &tf_session, &tf_memmapped_env);
388 load_status = LoadModel(model_file_name, model_file_type, &tf_session);
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| D | meta_graph_transform_test.py | 29 from tensorflow.python.client import session as tf_session unknown
53 with tf_session.Session(''):
62 with tf_session.Session(''):
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | graph_actions.py | 41 from tensorflow.python.client import session as tf_session unknown
683 with tf_session.Session('') as session:
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | estimator.py | 59 from tensorflow.python.client import session as tf_session unknown
1415 with tf_session.Session('') as session:
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 67 from tensorflow.python.client import session as tf_session unknown
161 tpu_session = tf_session.Session(target=master, config=config)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 38 from tensorflow.python.client import session as tf_session unknown
532 with tf_session.Session(
|
| /external/tensorflow/tensorflow/python/ |
| D | BUILD | 4327 "client/tf_session.i",
|