Searched defs:py_context (Results 1 – 2 of 2) sorted by relevance
98 TFE_Context* GetContextHandle(PyObject* py_context) { in GetContextHandle()462 PyObject* py_context = GetPyEagerContext(); in EagerTensor_init() local871 PyObject* py_context = GetPyEagerContext(); in EagerTensorFromHandle() local1035 PyObject* py_context = GetPyEagerContext(); in TFE_Py_TensorShapeSlice() local
4244 PyObject* TFE_Py_SetEagerContext(PyObject* py_context) {4258 PyObject* py_context = PyWeakref_GET_OBJECT(weak_eager_context); local