Factor out tiling operations out of tiled_dot_general
Currently tiled_dot_general does multiple things: tiling operation, dot_general, and detile-ing. So tiling information like the final shape after the tiling, or tiled_map isn't exposed. This CL aims to factor out the tiling operation parts by having additional functions, which returns the intermediate tiling information available to the callers.
Factor out tiling operations out of
tiled_dot_general
Currently tiled_dot_general does multiple things: tiling operation, dot_general, and detile-ing. So tiling information like the final shape after the tiling, or tiled_map isn't exposed. This CL aims to factor out the tiling operation parts by having additional functions, which returns the intermediate tiling information available to the callers.