Home
last modified time | relevance | path

Searched defs:host_context (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/runtime_fallback/util/
Dtensor_util_test.cc44 auto host_context = CreateTestHostContext(); in TEST() local
61 auto host_context = CreateTestHostContext(); in TEST() local
Dattr_util_test.cc286 auto host_context = CreateTestHostContext(); in TEST() local
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt_distributed_impl.cc117 HostContext* host_context) in DistributedManagerContextImpl()
226 CreateDistributedManagerContext(HostContext* host_context) { in CreateDistributedManagerContext()
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dtfrt_op_kernel_test.cc206 auto host_context = CreateTestHostContext(1); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.cc64 tfrt::HostContext* host_context) { in GetOrCreateReadyEvent()
114 auto host_context = std::make_unique<tfrt::HostContext>( in GetTfrtCpuClient() local
1345 auto* host_context = client_->GetHostContext(); in ExecuteHelper() local
/external/tensorflow/tensorflow/core/tfrt/saved_model/
Dsaved_model.cc434 auto* host_context = core_runtime->GetHostContext(); in OpenBefFile() local