Lines Matching refs:thunk
50 #include "tensorflow/compiler/xla/service/gpu/thunk.h"
117 // Updates the launch dimensions in "thunk" and annotate the launch dimensions
119 // Precondition: "thunk" must be a KernelThunk.
120 void UpdateLaunchDimensions(const LaunchDimensions& launch_dims, Thunk* thunk,
122 CHECK(Thunk::Kind::kKernel == thunk->kind());
123 KernelThunk* kernel_thunk = static_cast<KernelThunk*>(thunk);
432 std::unique_ptr<ConvolutionThunk> thunk;
434 thunk = MakeUnique<ConvolutionThunk>(
447 thunk = MakeUnique<ConvolutionThunk>(
460 thunk = MakeUnique<ConvolutionThunk>(
477 thunk_sequence_->emplace_back(std::move(thunk));
495 // array. The specialized version requires a initializer thunk that
501 std::vector<std::unique_ptr<Thunk>> thunks;
569 // Set up kernel thunk and fused ir emitter.
588 CHECK(Thunk::Kind::kKernel == LastThunk()->kind());
1115 CHECK(LastThunk()->kind() == Thunk::Kind::kSequential);
1285 CHECK(LastThunk()->kind() == Thunk::Kind::kSequential);
1555 CHECK(LastThunk()->kind() == Thunk::Kind::kSequential);
1666 // array. The specialized version requires an initializer thunk that
1671 std::vector<std::unique_ptr<Thunk>> thunks;
1743 std::vector<std::unique_ptr<Thunk>> thunks;
1763 // implemented by the initializer thunk generated earlier, so this function
2072 std::unique_ptr<Thunk> IrEmitterUnnested::BuildKernelThunk(
2168 std::unique_ptr<Thunk> IrEmitterUnnested::BuildHostToDeviceCopyThunk(
2181 std::unique_ptr<Thunk> IrEmitterUnnested::BuildDeviceToDeviceCopyThunk(
2193 std::unique_ptr<Thunk> IrEmitterUnnested::BuildInfeedThunk(
2210 std::unique_ptr<Thunk> IrEmitterUnnested::BuildGemmThunk(
2254 std::unique_ptr<Thunk> IrEmitterUnnested::BuildFftThunk(
2265 KernelThunk* thunk) {
2294 thunk);
2387 std::unique_ptr<Thunk> IrEmitterUnnested::BuildWhileThunk(
2393 // Generate thunk sequence for while 'condition'.
2399 // Generate thunk sequence for while 'body'.
2411 std::unique_ptr<Thunk> IrEmitterUnnested::BuildForThunk(
2417 // Generate thunk sequence for while 'body' (will be used a For loop body).
2427 std::unique_ptr<Thunk> IrEmitterUnnested::BuildConditionalThunk(
2454 const llvm_ir::ElementGenerator& element_generator, KernelThunk* thunk) {
2462 UpdateLaunchDimensions(launch_dimensions, thunk,
2492 CHECK(Thunk::Kind::kKernel == LastThunk()->kind());
