Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dimmediate_execution_context.h142 virtual void SetThreadLocalDevicePlacementPolicy(
Dc_api.cc256 tensorflow::unwrap(ctx)->SetThreadLocalDevicePlacementPolicy( in TFE_ContextSetThreadLocalDevicePlacementPolicy()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.h193 void SetThreadLocalDevicePlacementPolicy( in SetThreadLocalDevicePlacementPolicy() function
197 GetEagerContext()->SetThreadLocalDevicePlacementPolicy(policy); in SetThreadLocalDevicePlacementPolicy()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h179 void SetThreadLocalDevicePlacementPolicy(
Dcontext.cc400 void EagerContext::SetThreadLocalDevicePlacementPolicy( in SetThreadLocalDevicePlacementPolicy() function in tensorflow::EagerContext