Searched refs:kernel_class_name (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 1109 : def(d), kernel_class_name(c), factory(std::move(f)) {} in KernelRegistration()
1112 const string kernel_class_name; member
1242 StringPiece kernel_class_name, in InitInternal() argument
1262 KernelRegistration(*kernel_def, kernel_class_name, std::move(factory))); in InitInternal()
1385 const KernelDef** def, string* kernel_class_name) { in FindKernelDef() argument
1408 if (kernel_class_name != nullptr) *kernel_class_name = reg->kernel_class_name; in FindKernelDef()
1413 const KernelDef** def, string* kernel_class_name) { in FindKernelDef() argument
1417 AttrSlice(&node_def.attr()), def, kernel_class_name); in FindKernelDef()
|
| D | op_kernel.h | 1519 const KernelDef** def, string* kernel_class_name);
1525 const KernelDef** def, string* kernel_class_name);
1557 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() argument
1562 InitInternal(kernel_def, kernel_class_name, std::move(factory)); in OpKernelRegistrar()
1568 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() argument
1573 InitInternal(kernel_def, kernel_class_name, in OpKernelRegistrar()
1588 void InitInternal(const KernelDef* kernel_def, StringPiece kernel_class_name,
|
| D | op_kernel_test.cc | 637 string kernel_class_name; local
639 FindKernelDef(device_type, def, &kernel_def, &kernel_class_name);
641 return kernel_class_name;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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()
|