Home
last modified time | relevance | path

Searched refs:TFE_ContextDevicePlacementPolicy (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api.h65 typedef enum TFE_ContextDevicePlacementPolicy { enum
76 } TFE_ContextDevicePlacementPolicy; typedef
85 TFE_ContextOptions*, TFE_ContextDevicePlacementPolicy);
114 TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy);
118 TF_CAPI_EXPORT extern TFE_ContextDevicePlacementPolicy
Dc_api_internal.h58 TFE_ContextDevicePlacementPolicy device_placement_policy{
Dc_api.cc692 TFE_ContextOptions* options, TFE_ContextDevicePlacementPolicy policy) { in TFE_ContextOptionsSetDevicePlacementPolicy()
882 TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy) { in TFE_ContextSetThreadLocalDevicePlacementPolicy()
890 extern TFE_ContextDevicePlacementPolicy TFE_ContextGetDevicePlacementPolicy( in TFE_ContextGetDevicePlacementPolicy()
892 return static_cast<TFE_ContextDevicePlacementPolicy>( in TFE_ContextGetDevicePlacementPolicy()
/external/tensorflow/tensorflow/java/src/main/native/
Deager_session_jni.cc43 opts, static_cast<TFE_ContextDevicePlacementPolicy>(dpp)); in Java_org_tensorflow_EagerSession_allocate()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc432 [](py::handle& ctx, TFE_ContextDevicePlacementPolicy policy) { in PYBIND11_MODULE()
1066 py::enum_<TFE_ContextDevicePlacementPolicy>( in PYBIND11_MODULE()