Searched refs:SetLaunchDimensions (Results 1 – 3 of 3) sorted by relevance
58 void SetLaunchDimensions(const LaunchDimensions& launch_dims);
65 void KernelThunk::SetLaunchDimensions(const LaunchDimensions& launch_dims) { in SetLaunchDimensions() function in xla::gpu::KernelThunk
162 kernel_thunk->SetLaunchDimensions(launch_dims); in UpdateLaunchDimensions()