| /external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/ |
| D | cpu_registration.cc | 33 static void RegisterCpuOpHandler(CoreRuntime* core_runtime, in RegisterCpuOpHandler() argument
40 auto cpu = core_runtime->GetHostContext() in RegisterCpuOpHandler()
44 tensorflow::tfd::CreateRuntimeFallbackOpHandler(core_runtime, in RegisterCpuOpHandler()
49 core_runtime, std::move(cpu), expected_fallback_op_handler.get()); in RegisterCpuOpHandler()
62 core_runtime->RegisterOpHandler(device->name(), in RegisterCpuOpHandler()
68 core_runtime->RegisterOpHandler("cpu", expected_cpu_op_handler.get()); in RegisterCpuOpHandler()
|
| D | BUILD | 24 "//tensorflow/core/tfrt/eager:core_runtime",
30 "@tf_runtime//backends/cpu:core_runtime",
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/ |
| D | gpu_registration.cc | 33 static void RegisterGpuOpHandler(CoreRuntime* core_runtime, in RegisterGpuOpHandler() argument
54 core_runtime->GetHostContext()); in RegisterGpuOpHandler()
62 tensorflow::tfd::CreateRuntimeFallbackOpHandler(core_runtime, in RegisterGpuOpHandler()
67 ::tfrt::gpu::CreateGpuOpHandler(core_runtime, std::move(gpu.get()), in RegisterGpuOpHandler()
71 core_runtime->RegisterOpHandler(device->name(), in RegisterGpuOpHandler()
76 core_runtime->RegisterOpHandler("gpu", expected_gpu_op_handler.get()); in RegisterGpuOpHandler()
|
| D | BUILD | 37 "//tensorflow/core/tfrt/eager:core_runtime",
|
| /external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| D | BUILD | 23 "//tensorflow/core/tfrt/eager:core_runtime",
24 "@tf_runtime//:core_runtime",
27 "@tf_runtime//backends/cpu:core_runtime",
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | utils.cc | 126 auto* core_runtime = runtime.core_runtime(); in CreateBefFileFromBefBuffer() local
127 DCHECK(core_runtime); in CreateBefFileFromBefBuffer()
128 auto* host_context = core_runtime->GetHostContext(); in CreateBefFileFromBefBuffer()
|
| D | BUILD | 52 "@tf_runtime//:core_runtime",
96 "@tf_runtime//:core_runtime",
119 "@tf_runtime//:core_runtime",
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | BUILD | 28 "@tf_runtime//:core_runtime",
50 "@tf_runtime//:core_runtime",
74 "@tf_runtime//:core_runtime",
100 "@tf_runtime//:core_runtime",
123 "@tf_runtime//:core_runtime",
154 "@tf_runtime//:core_runtime",
157 "@tf_runtime//backends/cpu:core_runtime",
298 "@tf_runtime//:core_runtime",
325 "@tf_runtime//:core_runtime",
367 "@tf_runtime//:core_runtime",
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_selector.cc | 32 EagerOpHandlerSelector::EagerOpHandlerSelector(CoreRuntime* core_runtime, in EagerOpHandlerSelector() argument
36 : core_runtime_(core_runtime), in EagerOpHandlerSelector()
38 cpu_device_(core_runtime->GetHostContext()->GetHostDevice()), in EagerOpHandlerSelector()
|
| D | op_handler_registry.cc | 31 void RegisterOpHandlers(CoreRuntime* core_runtime, in RegisterOpHandlers() argument
35 fn(core_runtime, resource_context, device_mgr); in RegisterOpHandlers()
|
| D | op_handler_registry.h | 35 using OpHandlerRegistrationFn = void (*)(CoreRuntime* core_runtime,
40 void RegisterOpHandlers(CoreRuntime* core_runtime,
|
| D | op_handler_selector.h | 44 EagerOpHandlerSelector(CoreRuntime* core_runtime, EagerContext* eager_context,
|
| D | BUILD | 2 # core_runtime libraries.
|
| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | BUILD | 39 "@tf_runtime//:core_runtime",
102 "@tf_runtime//:core_runtime",
159 "@tf_runtime//:core_runtime",
181 "@tf_runtime//:core_runtime",
206 "@tf_runtime//:core_runtime",
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | BUILD | 73 ":core_runtime",
139 ":core_runtime",
198 name = "core_runtime",
199 actual = "//tensorflow/core/tfrt/eager/core_runtime:core_runtime_lib",
254 "@tf_runtime//backends/cpu:core_runtime",
|
| /external/tensorflow/tensorflow/core/tfrt/runtime/ |
| D | runtime.h | 63 tfrt::CoreRuntime* core_runtime() const { return core_runtime_.get(); } in core_runtime() function
92 explicit Runtime(std::unique_ptr<tfrt::CoreRuntime> core_runtime,
|
| D | runtime.cc | 158 Runtime::Runtime(std::unique_ptr<tfrt::CoreRuntime> core_runtime, in Runtime() argument
160 : core_runtime_(std::move(core_runtime)), work_queue_(work_queue) { in Runtime()
|
| D | BUILD | 44 "@tf_runtime//backends/cpu:core_runtime",
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | BUILD | 87 "@tf_runtime//:core_runtime",
92 "@tf_runtime//backends/cpu:core_runtime",
179 "@tf_runtime//:core_runtime",
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_sync.td | 21 include "tfrt/core_runtime/opdefs/corert_base.td"
22 include "tfrt/core_runtime/opdefs/corert_traits.td"
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | bef_thunk.cc | 85 tfrt::CoreRuntime* core_runtime; member
226 tfrt::HostContext* host = runtime_and_queue.core_runtime->GetHostContext(); in CreateBefThunk()
338 runtime_and_queue.core_runtime->GetHostContext(), in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| D | tfrt_fallback_sync.td | 21 include "tfrt/core_runtime/opdefs/corert_base.td"
22 include "tfrt/core_runtime/opdefs/corert_traits.td"
|
| /external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| D | BUILD | 69 "@tf_runtime//:core_runtime",
|
| 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()
|