Searched refs:launch_bound (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | launch_dimensions.h | 54 int64 launch_bound() const { in launch_bound() function
|
| D | parallel_loop_emitter.cc | 163 int64 total_threads = launch_dimensions_.launch_bound(); in EmitLoop()
|
| D | ir_emitter_unnested.cc | 933 pad_to_static, launch_dimensions.launch_bound(), &b_))); in EmitPadToStaticFromMlir()
1057 launch_dimensions.launch_bound(), &b_))); in EmitSliceToDynamicFromMlir()
1847 fusion, launch_dimensions.launch_bound(), &b_); in EmitLoopFusionFromMlir()
1929 fusion_op, launch_dimensions.launch_bound(), &b_))); in HandleFusion()
2241 select_and_scatter_op, launch_dimensions.launch_bound(), &b_); in EmitSelectAndScatterFromMlir()
2730 desc.get_index_type(launch_dimensions.launch_bound())); in EmitScatter()
4837 GetIndexTypeForKernelFromMlir(op, launch_dimensions.launch_bound(), &b_); in EmitHlo021Tile()
5431 unnested_hlo, launch_dimensions.launch_bound(), &b_); in EmitIRForReduction()
5822 fusion, launch_dimensions.launch_bound(), &b_)); in EmitInputFusibleNonStridedSlices()
|