Searched refs:kernel_thunk (Results 1 – 2 of 2) sorted by relevance
150 KernelThunk* kernel_thunk = static_cast<KernelThunk*>(thunk); in UpdateLaunchDimensions() local151 kernel_thunk->SetLaunchDimensions(launch_dims); in UpdateLaunchDimensions()158 llvm_module->getFunction(kernel_thunk->kernel_name().c_str()); in UpdateLaunchDimensions()1989 std::unique_ptr<KernelThunk> kernel_thunk = in BuildInitializerThunk() local1994 UpdateLaunchDimensions(launch_dimensions, kernel_thunk.get(), in BuildInitializerThunk()2028 return {std::move(kernel_thunk)}; in BuildInitializerThunk()2252 std::unique_ptr<KernelThunk> kernel_thunk = BuildKernelThunk( in EmitTargetElementLoop() local2255 EmitTargetElementLoopInThunk(hlo, element_generator, kernel_thunk.get()); in EmitTargetElementLoop()2256 thunk_sequence_->emplace_back(std::move(kernel_thunk)); in EmitTargetElementLoop()3451 std::unique_ptr<KernelThunk> kernel_thunk = in CheckAndEmitHloWithTile021() local[all …]
339 "kernel_thunk.cc",359 "kernel_thunk.h",