Home
last modified time | relevance | path

Searched refs:kernel_body_generator (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.h268 const std::function<void(ArgumentVector)>& kernel_body_generator);
276 kernel_body_generator) { in EmitAndCallOutlinedKernel() argument
279 kernel_body_generator(args[0], args[1], in EmitAndCallOutlinedKernel()
289 llvm::Value*)>& kernel_body_generator) { in EmitAndCallOutlinedKernel() argument
293 kernel_body_generator(args[0], args[1], args[2], args[3]); in EmitAndCallOutlinedKernel()
Dkernel_support_library.cc77 kernel_body_generator) { in EmitAndCallOutlinedKernel()
130 kernel_body_generator(arg_values); in EmitAndCallOutlinedKernel()