Searched refs:dims_in_blocks (Results 1 – 1 of 1) sorted by relevance
4698 std::vector<int64> dims_in_blocks = { in EmitTilingKernel() local4720 PRED /*arbitrary*/, dims_in_blocks), in EmitTilingKernel()4727 return IrArray::Index(multidim, dims_in_blocks, index_ty); in EmitTilingKernel()4735 b_.CreateICmpEQ(block_coords[i], constant(dims_in_blocks[i] - 1)); in EmitTilingKernel()4737 dims_in_elems[i] - (dims_in_blocks[i] - 1) * tile_size_for_dim; in EmitTilingKernel()4768 llvm::Value* last_block_for_dim = constant(dims_in_blocks[kDimZ] - 1); in EmitTilingKernel()4771 (dims_in_blocks[kDimZ] - 1) * mapping_scheme.GetTileSizeZ()); in EmitTilingKernel()