Searched defs:kernel_class_name (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | tfrt_op_kernel.cc | 269 void TFRTOpKernelFactories::RegisterFactory(StringPiece kernel_class_name, in RegisterFactory()
275 Status ValidKernelAttr(StringPiece kernel_class_name, in ValidKernelAttr()
298 StringPiece kernel_class_name, in CreateKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.cc | 1136 const string kernel_class_name; member
1322 StringPiece kernel_class_name, in InitInternal()
1476 const KernelDef** def, string* kernel_class_name) { in FindKernelDef()
1514 const KernelDef** def, string* kernel_class_name) { in FindKernelDef()
|
| H A D | op_kernel_test_base.h | 154 string kernel_class_name; variable
|
| H A D | op_kernel.h | 1510 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar()
1518 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_op_registry.cc | 48 string kernel_class_name; in LaunchOpHasKernelForDevice() local
|