Home
last modified time | relevance | path

Searched refs:TFE_ContextSetSoftDevicePlacement (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental.h575 TF_CAPI_EXPORT void TFE_ContextSetSoftDevicePlacement(TFE_Context* ctx,
Dc_api_experimental.cc641 void TFE_ContextSetSoftDevicePlacement(TFE_Context* ctx, unsigned char enable, in TFE_ContextSetSoftDevicePlacement() function
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc769 TFE_ContextSetSoftDevicePlacement(tensorflow::InputTFE_Context(ctx), enable, in PYBIND11_MODULE()
775 TFE_ContextSetSoftDevicePlacement(tensorflow::InputTFE_Context(ctx), enable, in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/eager/
Dcontext.py1651 pywrap_tfe.TFE_ContextSetSoftDevicePlacement(self._handle, enable)