Searched defs:kernel_builder (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | kernels.cc | 78 void AddTypeConstraint(TF_KernelBuilder* kernel_builder, const char* attr_name, in AddTypeConstraint()
110 void TF_KernelBuilder_TypeConstraint(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_TypeConstraint()
118 void TF_KernelBuilder_HostMemory(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_HostMemory()
123 void TF_KernelBuilder_Priority(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_Priority()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 1440 #define TF_EXTRACT_KERNEL_NAME(m, kernel_builder, ...) \ argument
1475 #define REGISTER_KERNEL_BUILDER_IMPL(kernel_builder, is_system_kernel, ...) \ argument
1479 #define REGISTER_KERNEL_BUILDER(kernel_builder, ...) \ argument
1486 #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
|
| /external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/ |
| D | passes.cc | 254 mlir::OpBuilder kernel_builder(gpu_module.body()); in runOnFunction() local
|