Searched refs:InputTFE_Context (Results 1 – 1 of 1) sorted by relevance
91 TFE_Context* InputTFE_Context(const py::handle& ctx) { in InputTFE_Context() function231 TFE_Context* ctx = tensorflow::InputTFE_Context(context); in InputTFE_OutputTensorHandles()249 TFE_Context* ctx = tensorflow::InputTFE_Context(context); in InputTFE_OutputTensorHandles()306 reinterpret_cast<ImmediateExecutionContext*>(InputTFE_Context(ctx))); in InputTFE_OutputTensorHandles()531 tensorflow::InputTFE_Context(ctx)); in PYBIND11_MODULE()629 TFE_DeleteContext(tensorflow::InputTFE_Context(o)); in PYBIND11_MODULE()636 auto output = TFE_ContextListDevices(tensorflow::InputTFE_Context(o), in PYBIND11_MODULE()643 TFE_HostAddressSpace(tensorflow::InputTFE_Context(o), &buf); in PYBIND11_MODULE()648 TFE_ContextAddFunction(tensorflow::InputTFE_Context(ctx), func, in PYBIND11_MODULE()656 TFE_ContextAddFunctionDef(tensorflow::InputTFE_Context(ctx), in PYBIND11_MODULE()[all …]