Home
last modified time | relevance | path

Searched refs:TFE_ContextDevicePlacementPolicy (Results 1 – 6 of 6) 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.h33 TFE_ContextDevicePlacementPolicy device_placement_policy{
Dc_api.cc109 TFE_ContextOptions* options, TFE_ContextDevicePlacementPolicy policy) { in TFE_ContextOptionsSetDevicePlacementPolicy()
252 TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy) { in TFE_ContextSetThreadLocalDevicePlacementPolicy()
260 extern TFE_ContextDevicePlacementPolicy TFE_ContextGetDevicePlacementPolicy( in TFE_ContextGetDevicePlacementPolicy()
262 return static_cast<TFE_ContextDevicePlacementPolicy>( in TFE_ContextGetDevicePlacementPolicy()
Dc_api_test.cc397 TFE_ContextDevicePlacementPolicy global_policy, in TensorHandleSilentCopy()
398 TFE_ContextDevicePlacementPolicy thread_policy, in TensorHandleSilentCopy()
/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.cc718 [](py::handle& ctx, TFE_ContextDevicePlacementPolicy policy) { in PYBIND11_MODULE()
1492 py::enum_<TFE_ContextDevicePlacementPolicy>( in PYBIND11_MODULE()