Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.h210 void SetThreadLocalDevicePlacementPolicy( in SetThreadLocalDevicePlacementPolicy() function
214 GetEagerContext()->SetThreadLocalDevicePlacementPolicy(policy); in SetThreadLocalDevicePlacementPolicy()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dimmediate_execution_context.h148 virtual void SetThreadLocalDevicePlacementPolicy(
H A Dc_api.cc258 tensorflow::unwrap(ctx)->SetThreadLocalDevicePlacementPolicy( in TFE_ContextSetThreadLocalDevicePlacementPolicy()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcontext.h188 void SetThreadLocalDevicePlacementPolicy(
H A Dcontext.cc503 void EagerContext::SetThreadLocalDevicePlacementPolicy( in SetThreadLocalDevicePlacementPolicy() function in tensorflow::EagerContext