Searched defs:index_ty (Results 1 – 4 of 4) sorted by relevance
267 void SetIndexType(llvm::Type* index_ty) { in SetIndexType()
57 explicit Index(llvm::Type* index_ty) : index_type_(index_ty) { in Index()
3943 llvm::Type* index_ty, in GetStartOffsetX()3975 llvm::Type* index_ty = tile_width->getType(); in UnrollInnerTileLoop() local4006 llvm::Type* index_ty = tile_width->getType(); in EmitTile() local4378 llvm::Type* index_ty, mlir::Operation* unnested_hlo, in EmitEpilogueForReduction()4670 llvm::Type* index_ty) { in EmitThreadId()4681 int64 threads_per_block, llvm::Type* index_ty, int64 num_threads_x) { in EmitThreadIdInfo()4695 const KernelMappingScheme& mapping_scheme, llvm::Type* index_ty, in EmitTilingKernel()4745 llvm::Type* index_ty = block_coords.GetType(); in EmitTilingKernel() local5430 llvm::Type* index_ty = GetIndexTypeForKernelFromMlir( in EmitIRForReduction() local
1973 RankedTensorType index_ty = index.getType().dyn_cast<RankedTensorType>(); in Verify() local