| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| H A D | op_handler_selector_test.cc | 409 OpHandler* op_handler = nullptr; in TEST_F() local
438 OpHandler* op_handler = nullptr; in TEST_F() local
459 OpHandler* op_handler = nullptr; in TEST_F() local
473 OpHandler* op_handler = nullptr; in TEST_F() local
486 OpHandler* op_handler = nullptr; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| H A D | op_handler_selector.cc | 49 const ImmediateExecutionOperation& op, OpHandler** op_handler) { in SelectFromArguments()
89 OpHandler** op_handler) { in SelectFromNodeDef()
|
| /aosp_15_r20/external/aws-crt-java/src/native/ |
| H A D | custom_key_op_handler.c | 36 …struct aws_jni_custom_key_op_handler *op_handler = (struct aws_jni_custom_key_op_handler *)key_op_… in s_aws_custom_key_op_handler_perform_operation() local
113 …struct aws_jni_custom_key_op_handler *op_handler = (struct aws_jni_custom_key_op_handler *)key_op_… in s_aws_custom_key_op_handler_destroy() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | kernel_fallback_kernels.cc | 94 tfrt::Result<tfrt::OpHandler*> op_handler, in CreateKernelFallbackOpHandlerKernel()
105 tfrt::Argument<tfrt::OpHandler*> op_handler, in AddKernelFallbackImplicitConversionKernel()
|
| H A D | kernel_fallback_op_handler.cc | 317 std::unique_ptr<KernelFallbackOpHandler> op_handler( in CreateKernelFallbackOpHandler() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A 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()
|
| H A D | op_cache.cc | 26 string_view op_name, OpHandler* op_handler, string_view device_name, in GetOrAddOp()
|
| H A D | op_cache_test.cc | 128 auto* op_handler = in TEST_F() local
|
| H A D | c_api_tfrt.cc | 1185 const tensorflow::ImmediateExecutionOperation& op, OpHandler** op_handler) { in SelectOpHandlerFromArguments()
1191 OpHandler** op_handler) { in SelectOpHandlerFromNodeDef()
1441 OpHandler* op_handler = nullptr; in Initialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | kernel_utils.cc | 98 tfrt::OpHandler* op_handler = nullptr; in GetFallbackOp() local
|
| H A D | runtime_fallback_batch_kernels.cc | 149 auto* op_handler = runtime->GetOpHandler(kKernelFallbackOpHandlerName); in BatchFunctionFallback() local
|
| H A D | runtime_fallback_op_handler.cc | 289 std::unique_ptr<RuntimeFallbackOpHandler> op_handler( in CreateRuntimeFallbackOpHandler() local
|
| H A D | runtime_fallback_kernels.cc | 1176 Argument<tfrt::OpHandler*> op_handler, const ExecutionContext& exec_ctx) { in AddRuntimeFallbackImplicitConversionKernel()
1189 void CreateRuntimeFallbackOpHandlerKernel(Result<tfrt::OpHandler*> op_handler, in CreateRuntimeFallbackOpHandlerKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/ |
| H A D | runtime.cc | 67 tfrt::OpHandler* op_handler = nullptr; in InitializeOpHandlers() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | tf_to_tfrt.cc | 321 auto op_handler = in ConvertToCoreRTExecuteOp() local
859 auto op_handler = corert_converter_.ConvertOpHandler( in matchAndRewrite() local
|