Searched refs:py_eager_context (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_tensor_converter_wrapper.cc | 66 PythonTensorConverter MakePythonTensorConverter(py::handle py_eager_context) { in MakePythonTensorConverter() argument
67 Safe_PyObjectPtr tld = GetAttr_ThreadLocalData(py_eager_context.ptr()); in MakePythonTensorConverter()
79 GetAttr_ContextHandle(py_eager_context.ptr()); in MakePythonTensorConverter()
93 return PythonTensorConverter(py_eager_context.ptr(), ctx, device_name); in MakePythonTensorConverter()
|
| D | python_tensor_converter.h | 45 PythonTensorConverter(PyObject* py_eager_context, TFE_Context* ctx, in PythonTensorConverter() argument
47 : py_eager_context_(py_eager_context), in PythonTensorConverter()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe.h | 429 void MakeEagerContextThreadLocalData(PyObject* py_eager_context,
439 PyObject* py_eager_context);
453 void DestroyEagerContextThreadLocalData(PyObject* py_eager_context);
|
| D | pywrap_tfe_src.cc | 3575 PyObject* py_eager_context = local
3580 PyObject_GetAttrString(py_eager_context, "_context_handle");
3595 auto* tld = tensorflow::GetEagerContextThreadLocalData(py_eager_context);
4296 void MakeEagerContextThreadLocalData(PyObject* py_eager_context, argument
4305 if (eager_context_thread_local_data_defaults->count(py_eager_context) > 0) {
4312 (*eager_context_thread_local_data_defaults)[py_eager_context];
4320 PyObject* py_eager_context) { argument
4328 eager_context_thread_local_data_defaults->find(py_eager_context);
4341 (*eager_context_thread_local_data_map)[py_eager_context];
4379 void DestroyEagerContextThreadLocalData(PyObject* py_eager_context) { argument
[all …]
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 378 explicit EagerContextThreadLocalDataWrapper(py::handle py_eager_context, in EagerContextThreadLocalDataWrapper() argument
381 : py_eager_context_(py_eager_context.ptr()) { in EagerContextThreadLocalDataWrapper()
383 py_eager_context.ptr(), is_eager.ptr(), device_spec.ptr()); in EagerContextThreadLocalDataWrapper()
|