Searched refs:kernel_class_name (Results 1 – 6 of 6) 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() argument 271 factories_[std::string(kernel_class_name)].push_back(kernel_info); in RegisterFactory() 275 Status ValidKernelAttr(StringPiece kernel_class_name, in ValidKernelAttr() argument 284 "Kernel ", kernel_class_name, in ValidKernelAttr() 289 "Kernel ", kernel_class_name, " with type constraint ", attr_name, in ValidKernelAttr() 298 StringPiece kernel_class_name, in CreateKernel() argument 300 auto it = factories_.find(std::string(kernel_class_name)); in CreateKernel() 304 "Could not find kernel ", kernel_class_name, " in the registry.")); in CreateKernel() 309 Status s = ValidKernelAttr(kernel_class_name, op_kernel_construction, in CreateKernel()
|
H A D | tfrt_op_kernel.h | 272 void RegisterFactory(StringPiece kernel_class_name, 286 StringPiece kernel_class_name,
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | op_kernel_test_base.h | 154 string kernel_class_name; variable 156 FindKernelDef(device_type, def, &kernel_def, &kernel_class_name); 158 return kernel_class_name;
|
H A D | op_kernel.cc | 1133 : def(d), kernel_class_name(c), factory(std::move(f)) {} in KernelRegistration() 1136 const string kernel_class_name; member 1274 KernelRegistration(def_without_label, it->second.kernel_class_name, in SetupOrDisableJit() 1322 StringPiece kernel_class_name, in InitInternal() argument 1340 KernelRegistration(*kernel_def, kernel_class_name, std::move(factory))); in InitInternal() 1476 const KernelDef** def, string* kernel_class_name) { in FindKernelDef() argument 1509 if (kernel_class_name != nullptr) *kernel_class_name = reg->kernel_class_name; in FindKernelDef() 1514 const KernelDef** def, string* kernel_class_name) { in FindKernelDef() argument 1518 AttrSlice(&node_def.attr()), def, kernel_class_name); in FindKernelDef()
|
H A D | op_kernel.h | 1472 const KernelDef** def, std::string* kernel_class_name); 1478 const KernelDef** def, std::string* kernel_class_name); 1510 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() argument 1513 InitInternal(kernel_def, kernel_class_name, std::move(factory)); in OpKernelRegistrar() 1518 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() argument 1521 InitInternal(kernel_def, kernel_class_name, in OpKernelRegistrar() 1535 void InitInternal(const KernelDef* kernel_def, StringPiece kernel_class_name,
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | xla_op_registry.cc | 48 string kernel_class_name; in LaunchOpHasKernelForDevice() local 50 &kernel_class_name)); in LaunchOpHasKernelForDevice() 52 << " kernel_class_name: " << kernel_class_name; in LaunchOpHasKernelForDevice()
|