Searched defs:op_handler (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_selector.cc | 49 const ImmediateExecutionOperation& op, OpHandler** op_handler) { in SelectFromArguments()
89 OpHandler** op_handler) { in SelectFromNodeDef()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_kernels.cc | 94 tfrt::Result<tfrt::OpHandler*> op_handler, in CreateKernelFallbackOpHandlerKernel()
105 tfrt::Argument<tfrt::OpHandler*> op_handler, in AddKernelFallbackImplicitConversionKernel()
|
| D | kernel_fallback_op_handler.cc | 150 std::unique_ptr<KernelFallbackOpHandler> op_handler( in CreateKernelFallbackOpHandler() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | op_cache.h | 67 bool Contains(string_view op_name, OpHandler* op_handler, in Contains()
80 CacheKey(string_view op_name, OpHandler* op_handler, in CacheKey()
|
| D | op_cache.cc | 26 string_view op_name, OpHandler* op_handler, string_view device_name, in GetOrAddOp()
|
| D | op_cache_test.cc | 128 auto* op_handler = in TEST_F() local
|
| D | c_api_tfrt.cc | 1142 const tensorflow::ImmediateExecutionOperation& op, OpHandler** op_handler) { in SelectOpHandlerFromArguments()
1148 OpHandler** op_handler) { in SelectOpHandlerFromNodeDef()
1382 OpHandler* op_handler = nullptr; in Initialize() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| D | op_handler_selector_test.cc | 405 OpHandler* op_handler = nullptr; in TEST_F() local
434 OpHandler* op_handler = nullptr; in TEST_F() local
455 OpHandler* op_handler = nullptr; in TEST_F() local
469 OpHandler* op_handler = nullptr; in TEST_F() local
482 OpHandler* op_handler = nullptr; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | kernel_utils.cc | 98 tfrt::OpHandler* op_handler = nullptr; in GetFallbackOp() local
|
| D | runtime_fallback_batch_kernels.cc | 150 auto* op_handler = runtime->GetOpHandler(kKernelFallbackOpHandlerName); in BatchFunctionFallback() local
|
| D | runtime_fallback_op_handler.cc | 280 std::unique_ptr<RuntimeFallbackOpHandler> op_handler( in CreateRuntimeFallbackOpHandler() local
|
| D | runtime_fallback_kernels.cc | 1153 Argument<tfrt::OpHandler*> op_handler, const ExecutionContext& exec_ctx) { in AddRuntimeFallbackImplicitConversionKernel()
1166 void CreateRuntimeFallbackOpHandlerKernel(Result<tfrt::OpHandler*> op_handler, in CreateRuntimeFallbackOpHandlerKernel()
|
| /external/tensorflow/tensorflow/core/tfrt/runtime/ |
| D | runtime.cc | 84 tfrt::OpHandler* op_handler = nullptr; in InitializeOpHandlers() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | tf_to_tfrt.cc | 291 auto op_handler = in ConvertToCoreRTExecuteOp() local
829 auto op_handler = corert_converter_.ConvertOpHandler( in matchAndRewrite() local
|