InputBlockV2 - Extend test coverage #605
Description
Currently there are dozens of tests using InputBlock.
This task is about creating copies of those tests and replacing InputBlock by InputBlockV2, which over time will be the de-facto solution for InputBlock and flexible embedding table definition.
Associated to RMP #479, which you can check for more context