Searched refs:UpdateLaunchDimensions (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 158 void UpdateLaunchDimensions(const LaunchDimensions& launch_dims, Thunk* thunk, in UpdateLaunchDimensions() function
926 UpdateLaunchDimensions(launch_dimensions, kernel_thunk.get(), in EmitPadToStaticFromMlir()
1049 UpdateLaunchDimensions(launch_dimensions, kernel_thunk.get(), in EmitSliceToDynamicFromMlir()
1844 UpdateLaunchDimensions(launch_dimensions, kernel_thunk, in EmitLoopFusionFromMlir()
1921 UpdateLaunchDimensions(launch_dimensions, thunks.back().get(), in HandleFusion()
2044 UpdateLaunchDimensions(launch_dimensions, fusion_thunk.get(), in HandleFusion()
2417 UpdateLaunchDimensions( in EmitSelectAndScatterFromMlir()
2724 UpdateLaunchDimensions(launch_dimensions, thunk, in EmitScatter()
2997 UpdateLaunchDimensions(launch_dimensions, thunks.back().get(), in EmitSortFromMlir()
3675 UpdateLaunchDimensions(launch_dimensions, kernel_thunk.get(), in BuildInitializerThunkForMlir()
[all …]
|