Searched refs:ContextDevicePlacementPolicy (Results 1 – 6 of 6) sorted by relevance
58 enum ContextDevicePlacementPolicy { enum80 ContextDevicePlacementPolicy default_policy, bool async,85 ContextDevicePlacementPolicy default_policy, bool async,123 void SetThreadLocalDevicePlacementPolicy(ContextDevicePlacementPolicy policy);126 ContextDevicePlacementPolicy GetDevicePlacementPolicy();250 const ContextDevicePlacementPolicy policy_;255 std::unordered_map<std::thread::id, ContextDevicePlacementPolicy>
47 ContextDevicePlacementPolicy default_policy, in EagerContext()55 ContextDevicePlacementPolicy default_policy, in EagerContext()153 ContextDevicePlacementPolicy policy) { in SetThreadLocalDevicePlacementPolicy()158 ContextDevicePlacementPolicy EagerContext::GetDevicePlacementPolicy() { in GetDevicePlacementPolicy()
45 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in Prepare()
71 static_cast<tensorflow::ContextDevicePlacementPolicy>( in TFE_Context()
399 static_cast<tensorflow::ContextDevicePlacementPolicy>(policy)); in TFE_ContextSetThreadLocalDevicePlacementPolicy()
122 tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT, in CreateContext()