Commit 46afcd06 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Implement Dequantize Op in XLA. 

1. Only MIN_COMBINED mode is supported;
2. Reshape the output to [d0,..., dn * unpack_size] if input shape is [d0, ..., dn].
3. Only uint32 is supported for the input;
4. Output data type is bfloat16;
5. Only uint8 or uint16 is supported for the original unpacked input.

PiperOrigin-RevId: 225203930
parent da29d1c8
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment