| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.cc | 386 context_.GetHostContext()->Await(FormRef(async_tensor)); in DataType()
417 context_.GetHostContext()->Await(FormRef(async_tensor)); in TensorHandleStatus()
486 context_.GetHostContext()->Await(th.GetAsyncDevice().CopyRCRef()); in DeviceName()
504 context_.GetHostContext()->Await(th.GetAsyncDevice().CopyRCRef()); in DeviceType()
515 auto* host_ctx = context_.GetHostContext(); in Resolve()
567 context_.GetHostContext()->Await(th.GetAsyncMetadata().CopyRCRef()); in Metadata()
633 return MakeScalarTensor(value, GetHostContext()); in CreateInt64Scalar()
638 return MakeScalarTensor(value, GetHostContext()); in CreateUint64Scalar()
643 return MakeScalarTensor(value, GetHostContext()); in CreateInt32Scalar()
648 return MakeScalarTensor(value, GetHostContext()); in CreateFloatScalar()
[all …]
|
| D | tfrt_context.cc | 80 host_context_ = corert_->GetHostContext(); in TfrtContext()
109 tensorflow::tfrt_global::GlobalHostContext::Set(corert_->GetHostContext()); in TfrtContext()
|
| D | function_cache_test.cc | 227 auto& device = corert->GetHostContext()->GetHostDevice(); in TEST_P()
229 auto req_ctx = RequestContextBuilder(corert->GetHostContext(), in TEST_P()
235 [host = corert->GetHostContext()]( in TEST_P()
|
| D | tfrt_context.h | 85 HostContext* GetHostContext() { return host_context_; } in GetHostContext() function
|
| D | function_cache.cc | 102 HostContext* host_ctx = corert->GetHostContext(); in GetOrAddFunction()
|
| D | c_api_tfrt.h | 152 GetHostContext()->Quiesce(); in AsyncWait()
286 HostContext* GetHostContext();
|
| /external/tensorflow/tensorflow/core/tfrt/runtime/ |
| D | runtime.cc | 78 corert, corert->GetHostContext()->GetHostDeviceRef()); in InitializeOpHandlers()
84 auto cpu_device = corert->GetHostContext()->GetHostDeviceRef(); in InitializeOpHandlers()
112 default_device, gpu_ordinal, corert->GetHostContext()); in InitializeOpHandlers()
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model_testutil.h | 53 tfrt::HostContext* GetHostContext() const { in GetHostContext() function
54 return saved_model_->GetHostContext(); in GetHostContext()
|
| D | saved_model.cc | 202 auto* host = runtime.core_runtime()->GetHostContext(); in RunInitializers()
381 tfrt::HostContext* SavedModel::GetHostContext() const { in GetHostContext() function in tensorflow::tfrt_stub::SavedModel
382 return runtime_->core_runtime()->GetHostContext(); in GetHostContext()
615 ->GetHostContext()), in SavedModelImpl()
|
| D | saved_model.h | 143 tfrt::HostContext* GetHostContext() const;
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/ |
| D | cpu_registration.cc | 40 auto cpu = core_runtime->GetHostContext() in RegisterCpuOpHandler()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/ |
| D | gpu_registration.cc | 54 core_runtime->GetHostContext()); in RegisterGpuOpHandler()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_selector.cc | 38 cpu_device_(core_runtime->GetHostContext()->GetHostDevice()), in EagerOpHandlerSelector()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tfrt_cpu_pjrt_client.cc | 602 tfrt::MakeConstructedAsyncValueRef<CpuEvent>(GetHostContext()); in BufferFromHostLiteral()
617 tfrt::EnqueueWork(GetHostContext(), [literal, av = avs[0].CopyRef(), in BufferFromHostLiteral()
632 tfrt::EnqueueWork(GetHostContext(), [i, literal, av = avs[i].CopyRef(), in BufferFromHostLiteral()
803 client_->GetHostContext()->Await(av.CopyRCRef()); in Release()
875 client_->GetHostContext()->Await(av.CopyRCRef()); in logical_on_device_shape()
932 auto host_ctx = client_->GetHostContext(); in ToLiteral()
1094 src_definition_event.AndThen([host_ctx = client()->GetHostContext(), in CopyToDevice()
1309 auto* host_context = client_->GetHostContext(); in ExecuteHelper()
1684 tfrt::EnqueueWork(client_->GetHostContext(), [&, replica, partition, i] { in Execute()
|
| D | tfrt_cpu_pjrt_client.h | 226 tfrt::HostContext* GetHostContext() const { return host_ctx_.get(); } in GetHostContext() function
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | utils.cc | 128 auto* host_context = core_runtime->GetHostContext(); in CreateBefFileFromBefBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | bef_thunk.cc | 226 tfrt::HostContext* host = runtime_and_queue.core_runtime->GetHostContext(); in CreateBefThunk()
338 runtime_and_queue.core_runtime->GetHostContext(), in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| D | graph_executor.h | 122 options_.runtime->core_runtime()->GetHostContext()) {} in GraphExecutor()
|
| D | graph_executor.cc | 150 auto* host = runtime.core_runtime()->GetHostContext(); in GraphExecutionRunOnFunction()
528 auto* host = runtime().core_runtime()->GetHostContext(); in InitBef()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_op_handler.cc | 291 runtime, runtime->GetHostContext()->GetHostDeviceRef(), in CreateRuntimeFallbackOpHandler()
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/tests/ |
| D | saved_model_test.cc | 128 auto op_count = saved_model->GetHostContext() in TEST()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 127 tfrt_context->GetCoreRuntime()->GetHostContext())); in TFE_NewContext()
|