Searched defs:kernel_builder (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 78 void AddTypeConstraint(TF_KernelBuilder* kernel_builder, const char* attr_name, in AddTypeConstraint()
113 void TF_KernelBuilder_TypeConstraint(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_TypeConstraint()
121 void TF_KernelBuilder_HostMemory(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_HostMemory()
126 void TF_KernelBuilder_Priority(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_Priority()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 1389 #define TF_EXTRACT_KERNEL_NAME(m, kernel_builder, ...) \ argument
1424 #define REGISTER_KERNEL_BUILDER_IMPL(kernel_builder, is_system_kernel, ...) \ argument
1428 #define REGISTER_KERNEL_BUILDER(kernel_builder, ...) \ argument
1435 #define REGISTER_SYSTEM_KERNEL_BUILDER(kernel_builder, ...) \ argument
|
| D | op_kernel_test.cc | 1149 #define EXTRACT_KERNEL_NAME_TO_STRING_IMPL(name, kernel_builder, ...) name argument
1150 #define EXTRACT_KERNEL_NAME_TO_STRING(kernel_builder) \ argument
|