Home
last modified time | relevance | path

Searched refs:kernel_builder (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dpasses.cc254 mlir::OpBuilder kernel_builder(gpu_module.body()); in runOnFunction() local
257 auto new_kernel = kernel_builder.create<mlir::gpu::GPUFuncOp>( in runOnFunction()
259 kernel_builder.getFunctionType(operand_types, {})); in runOnFunction()
261 kernel_builder.getUnitAttr()); in runOnFunction()
294 kernel_builder.setInsertionPointToEnd(&new_kernel.body().front()); in runOnFunction()
295 kernel_builder.create<mlir::BranchOp>( in runOnFunction()
/external/tensorflow/tensorflow/c/
Dkernels.cc74 kernel_builder->cc_builder->TypeConstraint<type>(attr_name); \
78 void AddTypeConstraint(TF_KernelBuilder* kernel_builder, const char* attr_name, in AddTypeConstraint() argument
110 void TF_KernelBuilder_TypeConstraint(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_TypeConstraint() argument
115 tensorflow::AddTypeConstraint(kernel_builder, attr_name, dtype, status); in TF_KernelBuilder_TypeConstraint()
118 void TF_KernelBuilder_HostMemory(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_HostMemory() argument
120 kernel_builder->cc_builder->HostMemory(arg_name); in TF_KernelBuilder_HostMemory()
123 void TF_KernelBuilder_Priority(TF_KernelBuilder* kernel_builder, in TF_KernelBuilder_Priority() argument
125 kernel_builder->cc_builder->Priority(priority_number); in TF_KernelBuilder_Priority()
Dkernels.h110 TF_KernelBuilder* kernel_builder, const char* attr_name,
116 TF_KernelBuilder* kernel_builder, const char* arg_name);
120 TF_KernelBuilder* kernel_builder, int32_t priority_number);
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h1440 #define TF_EXTRACT_KERNEL_NAME(m, kernel_builder, ...) \ argument
1441 TF_EXTRACT_KERNEL_NAME_IMPL(m, TF_EXTRACT_KERNEL_NAME_##kernel_builder, \
1475 #define REGISTER_KERNEL_BUILDER_IMPL(kernel_builder, is_system_kernel, ...) \ argument
1476 TF_EXTRACT_KERNEL_NAME(REGISTER_KERNEL_BUILDER_IMPL_2, kernel_builder, \
1479 #define REGISTER_KERNEL_BUILDER(kernel_builder, ...) \ argument
1481 REGISTER_KERNEL_BUILDER_IMPL(kernel_builder, false, __VA_ARGS__)
1486 #define REGISTER_SYSTEM_KERNEL_BUILDER(kernel_builder, ...) \ argument
1489 REGISTER_KERNEL_BUILDER_IMPL(kernel_builder, true, __VA_ARGS__)
Dop_kernel_test.cc1149 #define EXTRACT_KERNEL_NAME_TO_STRING_IMPL(name, kernel_builder, ...) name argument
1150 #define EXTRACT_KERNEL_NAME_TO_STRING(kernel_builder) \ argument
1151 TF_EXTRACT_KERNEL_NAME(EXTRACT_KERNEL_NAME_TO_STRING_IMPL, kernel_builder)