Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental.h670 TF_CAPI_EXPORT extern void TFE_SetLogicalCpuDevices(TFE_Context* ctx,
Dc_api_experimental.cc758 void TFE_SetLogicalCpuDevices(TFE_Context* ctx, int num_cpus, in TFE_SetLogicalCpuDevices() function
/external/tensorflow/tensorflow/python/eager/
Dcontext.py1709 pywrap_tfe.TFE_SetLogicalCpuDevices(self._context_handle, num_cpus, prefix)
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc732 TFE_SetLogicalCpuDevices(tensorflow::InputTFE_Context(ctx), num_cpus, in PYBIND11_MODULE()