Searched defs:host_ctx (Results 1 – 13 of 13) sorted by relevance
50 /*static*/ void GlobalHostContext::Set(::tfrt::HostContext* host_ctx) { in Set()
117 HostContext* host_ctx) { in AddDummyTfrtDevices()
135 const tensorflow::Tensor& tf_tensor, HostContext* host_ctx) { in TFTensorToTFRTTensorHandle()
102 HostContext* host_ctx = corert->GetHostContext(); in GetOrAddFunction() local
66 const DistributedContextConfiguration& dist_config, HostContext* host_ctx) { in CreateServer()
515 auto* host_ctx = context_.GetHostContext(); in Resolve() local1040 auto* host_ctx = GetHostContext(); in CopyTensorHandleToDevice() local
70 auto* host_ctx = exec_ctx.host(); in ConvertRuntimeFallbackTensorToDenseGpuTensor() local
68 auto *host_ctx = exec_ctx.host(); in ConvertRuntimeFallbackTensorToStringHostTensor() local
261 auto* host_ctx = invocation.exec_ctx.host(); in MakeOp() local
512 HostContext* host_ctx, tfrt::ResourceContext* resource_context, in SetUpRequestContext()
176 std::unique_ptr<tfrt::HostContext> host_ctx) in TfrtCpuClient()910 tfrt::HostContext* host_ctx, in EnqueueWorkWhenReady()932 auto host_ctx = client_->GetHostContext(); in ToLiteral() local
350 HostCallbackContext* host_ctx = reinterpret_cast<HostCallbackContext*>(ctx); in HostCallbackTrampoline() local
203 HostCallbackContext* host_ctx = static_cast<HostCallbackContext*>(ctx); in HostCallbackTrampoline() local