Searched refs:GetIndexTypeForKernel (Results 1 – 1 of 1) sorted by relevance
293 llvm::Type* GetIndexTypeForKernel(const HloInstruction* hlo, int64 launch_size, in GetIndexTypeForKernel() function809 llvm::Type* index_type = GetIndexTypeForKernel( in HandleSelectAndScatter()1201 GetIndexTypeForKernel(scatter, launch_dimensions.launch_bound(), in EmitScatter()2195 GetIndexTypeForKernel(&hlo, launch_dimensions.launch_bound(), &b_)); in EmitTargetElementLoopInThunk()2212 GetIndexTypeForKernel( in EmitTargetElementLoopInThunk()3084 : GetIndexTypeForKernel(unnested_hlo, in EmitKernel()