Searched defs:kernel_builder (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | kernels.cc | 82 void AddTypeConstraint(TF_KernelBuilder* kernel_builder, const char* attr_name, in AddTypeConstraint()
117 void TF_KernelBuilder_TypeConstraint(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_TypeConstraint()
125 void TF_KernelBuilder_HostMemory(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_HostMemory()
130 void TF_KernelBuilder_Priority(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_Priority()
135 void TF_KernelBuilder_Label(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_Label()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.h | 1409 #define TF_EXTRACT_KERNEL_NAME(m, kernel_builder, ...) \ argument
1444 #define REGISTER_KERNEL_BUILDER_IMPL(kernel_builder, is_system_kernel, ...) \ argument
1448 #define REGISTER_KERNEL_BUILDER(kernel_builder, ...) \ argument
1455 #define REGISTER_SYSTEM_KERNEL_BUILDER(kernel_builder, ...) \ argument
|
| H A D | op_kernel_test.cc | 1046 #define EXTRACT_KERNEL_NAME_TO_STRING_IMPL(name, kernel_builder, ...) name argument
1047 #define EXTRACT_KERNEL_NAME_TO_STRING(kernel_builder) \ argument
|