Searched refs:function_library_runtime (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/ |
| H A D | op_kernel_runner.cc | 121 tensorflow::FunctionLibraryRuntime* function_library_runtime = nullptr; in Create() local
123 function_library_runtime = in Create()
127 TF_RETURN_IF_ERROR(CreateOpKernel(function_library_runtime, in Create()
129 return OpKernelRunner(device, function_library_runtime, std::move(op_kernel)); in Create()
134 tensorflow::FunctionLibraryRuntime* function_library_runtime, in OpKernelRunner() argument
137 function_library_runtime_(function_library_runtime), in OpKernelRunner()
|
| H A D | op_kernel_runner.h | 69 tensorflow::FunctionLibraryRuntime* function_library_runtime() const { in function_library_runtime() function
86 tensorflow::FunctionLibraryRuntime* function_library_runtime,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | op_kernel_runner.cc | 102 tensorflow::FunctionLibraryRuntime* function_library_runtime = nullptr; in Create() local
119 function_library_runtime = in Create()
124 TF_RETURN_IF_ERROR(CreateOpKernel(function_library_runtime, in Create()
126 return OpKernelRunner(device, function_library_runtime, std::move(op_kernel)); in Create()
131 tensorflow::FunctionLibraryRuntime* function_library_runtime, in OpKernelRunner() argument
134 function_library_runtime_(function_library_runtime), in OpKernelRunner()
|
| H A D | op_kernel_runner.h | 88 tensorflow::FunctionLibraryRuntime* function_library_runtime() const { in function_library_runtime() function
105 tensorflow::FunctionLibraryRuntime* function_library_runtime,
|
| H A D | kernel_fallback_utils.cc | 41 params.function_library = runner.function_library_runtime(); in SetUpParams()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | kernel_and_device_test.cc | 60 FunctionLibraryRuntime* function_library_runtime() const { return flr_; } in function_library_runtime() function in tensorflow::__anon79d852060111::TestEnv
116 KernelAndDeviceOp k(nullptr, false, env.function_library_runtime(), nullptr, in BM_KernelAndDeviceInit()
137 KernelAndDeviceOp k(nullptr, false, env.function_library_runtime(), nullptr, in BM_KernelAndDeviceRun()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | kernel.cc | 424 op_kernel_runner.function_library_runtime(); in ExecuteOpKernelRunner()
|