Searched refs:GetTileSizeFor (Results 1 – 2 of 2) sorted by relevance
108 CHECK_EQ(GetTileSizeFor(DimX) % num_threads_x_, 0);123 int64 GetTileSizeFor(int d) const { return tile_sizes_.at(d); } in GetTileSizeFor() function125 int64 GetTileSizeZ() const { return GetTileSizeFor(DimZ); } in GetTileSizeZ()126 int64 GetTileSizeX() const { return GetTileSizeFor(DimX); } in GetTileSizeX()127 int64 GetTileSizeY() const { return GetTileSizeFor(DimY); } in GetTileSizeY()
4732 int64 tile_size_for_dim = mapping_scheme.GetTileSizeFor(i); in EmitTilingKernel()4749 llvm::ConstantInt::get(index_ty, mapping_scheme.GetTileSizeFor(i)), in EmitTilingKernel()