Searched refs:kernel_thunk (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 161 KernelThunk* kernel_thunk = static_cast<KernelThunk*>(thunk); in UpdateLaunchDimensions() local
162 kernel_thunk->SetLaunchDimensions(launch_dims); in UpdateLaunchDimensions()
169 llvm_module->getFunction(kernel_thunk->kernel_name().c_str()); in UpdateLaunchDimensions()
822 auto kernel_thunk, in EmitPadToStaticFromMlir()
926 UpdateLaunchDimensions(launch_dimensions, kernel_thunk.get(), in EmitPadToStaticFromMlir()
934 thunk_sequence_.emplace_back(std::move(kernel_thunk)); in EmitPadToStaticFromMlir()
949 auto kernel_thunk, in EmitSliceToDynamicFromMlir()
1049 UpdateLaunchDimensions(launch_dimensions, kernel_thunk.get(), in EmitSliceToDynamicFromMlir()
1058 thunk_sequence_.emplace_back(std::move(kernel_thunk)); in EmitSliceToDynamicFromMlir()
1770 Thunk* kernel_thunk; in EmitLoopFusionFromMlir() local
[all …]
|
| D | ir_emitter_unnested.h | 502 void EmitHlo021Tile(mlir::Operation* op, Thunk* kernel_thunk,
|
| D | BUILD | 566 "kernel_thunk.cc",
589 "kernel_thunk.h",
|