Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api.h63 typedef enum TFE_ContextDevicePlacementPolicy { enum
74 } TFE_ContextDevicePlacementPolicy; typedef
82 TFE_ContextOptions*, TFE_ContextDevicePlacementPolicy);
109 TFE_Context*, TFE_ContextDevicePlacementPolicy);
113 TF_CAPI_EXPORT extern TFE_ContextDevicePlacementPolicy
Dc_api_internal.h62 TFE_ContextDevicePlacementPolicy policy{TFE_DEVICE_PLACEMENT_SILENT};
67 TFE_ContextDevicePlacementPolicy default_policy, bool async, in TFE_Context()
Dc_api.cc324 TFE_ContextOptions* options, TFE_ContextDevicePlacementPolicy policy) { in TFE_ContextOptionsSetDevicePlacementPolicy()
397 TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy) { in TFE_ContextSetThreadLocalDevicePlacementPolicy()
405 extern TFE_ContextDevicePlacementPolicy TFE_ContextGetDevicePlacementPolicy( in TFE_ContextGetDevicePlacementPolicy()
407 return static_cast<TFE_ContextDevicePlacementPolicy>( in TFE_ContextGetDevicePlacementPolicy()
/external/tensorflow/tensorflow/python/
Dpywrap_tfe.i193 %rename("%s") TFE_ContextDevicePlacementPolicy;