Searched refs:kernel_builder (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 1375 #define REGISTER_KERNEL_BUILDER(kernel_builder, ...) \ argument
1376 REGISTER_KERNEL_BUILDER_UNIQ_HELPER(__COUNTER__, kernel_builder, __VA_ARGS__)
1378 #define REGISTER_KERNEL_BUILDER_UNIQ_HELPER(ctr, kernel_builder, ...) \ argument
1379 REGISTER_KERNEL_BUILDER_UNIQ(ctr, kernel_builder, __VA_ARGS__)
1381 #define REGISTER_KERNEL_BUILDER_UNIQ(ctr, kernel_builder, ...) \ argument
1387 ? ::tensorflow::register_kernel::kernel_builder.Build() \
1398 #define REGISTER_SYSTEM_KERNEL_BUILDER(kernel_builder, ...) \ argument
1399 REGISTER_SYSTEM_KERNEL_BUILDER_UNIQ_HELPER(__COUNTER__, kernel_builder, \
1402 #define REGISTER_SYSTEM_KERNEL_BUILDER_UNIQ_HELPER(ctr, kernel_builder, ...) \ argument
1403 REGISTER_SYSTEM_KERNEL_BUILDER_UNIQ(ctr, kernel_builder, __VA_ARGS__)
[all …]
|