Add slight speed improvement by avoiding tiling and reshapes.#520

copybara-service[bot] wants to merge 1 commit intomaingoogle/edward2:mainfrom

cl_430084350google/edward2:cl_430084350Copy head branch name to clipboard