Home
last modified time | relevance | path

Searched refs:tf_session (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/client/
Dsession_list_devices_test.py24 from tensorflow.python.client import pywrap_tf_session 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)
Dsession.py32 from tensorflow.python.client import pywrap_tf_session as tf_session unknown
708 opts = tf_session.TF_NewSessionOptions(target=self._target, config=config)
711 self._session = tf_session.TF_NewSessionRef(self._graph._c_graph, opts)
714 tf_session.TF_DeleteSessionOptions(opts)
741 raw_device_list = tf_session.TF_SessionListDevices(self._session)
743 size = tf_session.TF_DeviceListCount(raw_device_list)
745 name = tf_session.TF_DeviceListName(raw_device_list, i)
746 device_type = tf_session.TF_DeviceListType(raw_device_list, i)
747 memory = tf_session.TF_DeviceListMemoryBytes(raw_device_list, i)
748 incarnation = tf_session.TF_DeviceListIncarnation(raw_device_list, i)
[all …]
Dtf_session_helper.cc48 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/tools/def_file_filter/
Dsymbols_pybind.txt77 [//tensorflow/core/common_runtime:device_factory] # device_lib, tfe, tf_session
82 [//tensorflow/core/common_runtime:session_options] # device_lib, tfe, tf_session
88 [//tensorflow/core/common_runtime:session_state] # tf_session
218 [//tensorflow/core:op_gen_lib] # tf_session
221 [//tensorflow/core/common_runtime:graph_constructor] # tf_session
224 [//tensorflow/c:python_api] # tf_session
237 [//tensorflow/python:numpy_lib] # tf_session
241 [//tensorflow/python:tf_session_helper] # tf_session
/external/tensorflow/tensorflow/python/
DBUILD5310 "//tensorflow/c:python_api", # tf_session
5315 "//tensorflow/core/common_runtime:graph_constructor", # tf_session
5317 "//tensorflow/core/common_runtime:session_options", # device_lib, tfe, tf_session
5318 "//tensorflow/core/common_runtime:session_state", # tf_session
5322 "//tensorflow/core:op_gen_lib", # tf_session
5351 ":tf_session_helper", # tf_session