Searched refs:GetIndexTypeForKernel (Results 1 – 1 of 1) sorted by relevance
325 llvm::Type* GetIndexTypeForKernel(const HloInstruction* hlo, in GetIndexTypeForKernel() function379 llvm::Type* GetIndexTypeForKernel(mlir::Operation* op, int64_t launch_size, in GetIndexTypeForKernel() function748 llvm::Type* index_ty = GetIndexTypeForKernel( in EmitPadToStatic()864 llvm::Type* index_ty = GetIndexTypeForKernel( in EmitSliceToDynamic()1773 GetIndexTypeForKernel(fusion, launch_dimensions.launch_bound(), &b_); in EmitLoopFusion()1951 llvm::Type* index_type = GetIndexTypeForKernel( in EmitSelectAndScatter()2250 return GetIndexTypeForKernel(scatter_op, launch_size, &b_); in EmitScatter()4209 llvm::Type* index_type = GetIndexTypeForKernel( in EmitHlo021Tile()4710 GetIndexTypeForKernel(fusion, in EmitIRForReduction()5140 GetIndexTypeForKernel( in EmitInputFusibleNonStridedSlices()[all …]