Searched refs:launch_bound (Results 1 – 2 of 2) sorted by relevance
50 int64 launch_bound() const { return block_count() * threads_per_block(); } in launch_bound() function
810 select_and_scatter, launch_dimensions.launch_bound(), &b_); in HandleSelectAndScatter()1201 GetIndexTypeForKernel(scatter, launch_dimensions.launch_bound(), in EmitScatter()2195 GetIndexTypeForKernel(&hlo, launch_dimensions.launch_bound(), &b_)); in EmitTargetElementLoopInThunk()2213 &hlo, launch_dimensions.launch_bound(), &b_))); in EmitTargetElementLoopInThunk()3085 launch_dimensions.launch_bound(), &b_); in EmitKernel()