[Pallas support] Add quant_blockwisely function, which quantizes tensor according to the BlockSpec of the inputs.#596
Open
copybara-service[bot] wants to merge 1 commit into
Open