search

google / aqt

Apache License 2.0
262 stars 27 forks source link

[Pallas support] Add `quant_blockwisely` function, which quantizes tensor according to the BlockSpec of the inputs. #596

Open copybara-service[bot] opened 6 months ago

copybara-service[bot] commented 6 months ago

[Pallas support] Add quant_blockwisely function, which quantizes tensor according to the BlockSpec of the inputs.