| /external/tensorflow/tensorflow/core/tfrt/runtime/ |
| D | runtime.h | 24 class CoreRuntime; variable
63 tfrt::CoreRuntime* core_runtime() const { return core_runtime_.get(); } in core_runtime()
92 explicit Runtime(std::unique_ptr<tfrt::CoreRuntime> core_runtime,
95 std::unique_ptr<tfrt::CoreRuntime> core_runtime_;
|
| D | runtime.cc | 46 tensorflow::Status InitializeOpHandlers(tfrt::CoreRuntime* corert) { in InitializeOpHandlers()
132 auto expected_core_runtime = tfrt::CoreRuntime::Create( in Create()
158 Runtime::Runtime(std::unique_ptr<tfrt::CoreRuntime> core_runtime, in Runtime()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_registry.h | 26 class CoreRuntime; variable
35 using OpHandlerRegistrationFn = void (*)(CoreRuntime* core_runtime,
40 void RegisterOpHandlers(CoreRuntime* core_runtime,
|
| D | op_handler_selector.h | 31 class CoreRuntime; variable
44 EagerOpHandlerSelector(CoreRuntime* core_runtime, EagerContext* eager_context,
62 CoreRuntime* core_runtime_;
|
| D | op_handler_registry.cc | 31 void RegisterOpHandlers(CoreRuntime* core_runtime, in RegisterOpHandlers()
|
| D | op_handler_selector.cc | 32 EagerOpHandlerSelector::EagerOpHandlerSelector(CoreRuntime* core_runtime, in EagerOpHandlerSelector()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | tfrt_context.h | 33 class CoreRuntime; variable
86 CoreRuntime* GetCoreRuntime() { return corert_.get(); } in GetCoreRuntime()
104 std::unique_ptr<CoreRuntime> corert_;
|
| D | tfrt_context.cc | 73 auto rt = CoreRuntime::Create(diag_handler, CreateMallocAllocator(), in TfrtContext()
|
| D | function_cache.h | 111 tensorflow::EagerContext* eager_ctx, tfrt::CoreRuntime* corert,
|
| D | function_cache.cc | 48 tensorflow::EagerContext* eager_ctx, tfrt::CoreRuntime* corert, in GetOrAddFunction()
|
| D | c_api_tfrt.h | 57 class CoreRuntime; variable
223 CoreRuntime* GetCoreRuntime();
|
| D | function_cache_test.cc | 216 ::tfrt::CoreRuntime* corert = tfrt_ctx->GetCoreRuntime(); in TEST_P()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_op_handler.cc | 72 explicit RuntimeFallbackOpHandler(tfrt::CoreRuntime* runtime,
79 tfrt::CoreRuntime* runtime, tfrt::string_view tf_device_name);
91 using tfrt::CoreRuntime;
287 tfrt::CoreRuntime* runtime, tfrt::string_view tf_device_name) { in CreateRuntimeFallbackOpHandler()
302 CoreRuntime* runtime, tfrt::RCReference<tfrt::Device> device, in RuntimeFallbackOpHandler()
|
| D | runtime_fallback_op_handler.h | 32 tfrt::CoreRuntime* runtime, tfrt::string_view tf_device_name);
|
| D | kernel_utils.cc | 93 auto* runtime = tfrt::CoreRuntime::GetFromHostContext(host); in GetFallbackOp()
|
| D | runtime_fallback_batch_kernels.cc | 145 auto* runtime = tfrt::CoreRuntime::GetFromHostContext(host); in BatchFunctionFallback()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_op_handler.cc | 55 explicit KernelFallbackOpHandler(tfrt::CoreRuntime* runtime,
58 tfrt::CoreRuntime* runtime, tfrt::RCReference<tfrt::Device> device);
70 using tfrt::CoreRuntime;
316 tfrt::CoreRuntime* runtime, tfrt::RCReference<tfrt::Device> device) { in CreateKernelFallbackOpHandler()
328 CoreRuntime* runtime, RCReference<tfrt::Device> device) in KernelFallbackOpHandler()
|
| D | kernel_fallback_op_handler.h | 29 tfrt::CoreRuntime* runtime, tfrt::RCReference<tfrt::Device> device);
|
| D | kernel_fallback_kernels.cc | 96 auto* runtime = tfrt::CoreRuntime::GetFromHostContext(exec_ctx.host()); in CreateKernelFallbackOpHandlerKernel()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/ |
| D | gpu_registration.cc | 31 using ::tfrt::CoreRuntime;
33 static void RegisterGpuOpHandler(CoreRuntime* core_runtime, in RegisterGpuOpHandler()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/function/ |
| D | function.h | 32 class CoreRuntime; variable
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/saved_model/ |
| D | saved_model.h | 34 class CoreRuntime; variable
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/ |
| D | cpu_registration.cc | 33 static void RegisterCpuOpHandler(CoreRuntime* core_runtime, in RegisterCpuOpHandler()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| D | op_handler_selector_test.cc | 323 static std::unique_ptr<CoreRuntime> CreateCoreRuntime() { in CreateCoreRuntime()
328 CoreRuntime::Create(diag_handler, tfrt::CreateMallocAllocator(), in CreateCoreRuntime()
392 std::unique_ptr<CoreRuntime> corert_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | bef_thunk.cc | 85 tfrt::CoreRuntime* core_runtime;
195 auto expected_core_runtime = tfrt::CoreRuntime::Create( in GetCoreRuntimeAndWorkQueue()
|