Searched refs:GetIndexTypeForKernelFromMlir (Results 1 – 1 of 1) sorted by relevance
434 llvm::Type* GetIndexTypeForKernelFromMlir(mlir::Operation* op, in GetIndexTypeForKernelFromMlir() function932 GetIndexTypeForKernelFromMlir( in EmitPadToStaticFromMlir()1055 .EmitLoop(ir_name, GetIndexTypeForKernelFromMlir( in EmitSliceToDynamicFromMlir()1846 llvm::Type* index_type = GetIndexTypeForKernelFromMlir( in EmitLoopFusionFromMlir()1928 GetIndexTypeForKernelFromMlir( in HandleFusion()2240 llvm::Type* index_type = GetIndexTypeForKernelFromMlir( in EmitSelectAndScatterFromMlir()2536 return GetIndexTypeForKernelFromMlir(scatter_op, launch_size, &b_); in EmitScatterFromMlir()4837 GetIndexTypeForKernelFromMlir(op, launch_dimensions.launch_bound(), &b_); in EmitHlo021Tile()5430 llvm::Type* index_ty = GetIndexTypeForKernelFromMlir( in EmitIRForReduction()5821 GetIndexTypeForKernelFromMlir( in EmitInputFusibleNonStridedSlices()