Searched refs:launch_dims (Results 1 – 5 of 5) sorted by relevance
36 const LaunchDimensions& launch_dims) { in operator <<() argument37 out << absl::StrFormat("[block: %d, thread: %d]", launch_dims.block_count(), in operator <<()38 launch_dims.threads_per_block()); in operator <<()
69 void KernelThunk::SetLaunchDimensions(const LaunchDimensions& launch_dims) { in SetLaunchDimensions() argument71 launch_dimensions_ = launch_dims; in SetLaunchDimensions()
58 const LaunchDimensions& launch_dims);
59 void SetLaunchDimensions(const LaunchDimensions& launch_dims);
147 void UpdateLaunchDimensions(const LaunchDimensions& launch_dims, Thunk* thunk, in UpdateLaunchDimensions() argument151 kernel_thunk->SetLaunchDimensions(launch_dims); in UpdateLaunchDimensions()162 launch_dims.threads_per_block()); in UpdateLaunchDimensions()