Commit fbbecc81 authored by Yunxing Dai's avatar Yunxing Dai Committed by TensorFlower Gardener
Browse files

Dynamic Padder 

- Change HloGetDimensionSizeRewriter to use DynamicDimensionInference,
  when trying to get the size of a dynamic dimension, GetDimensionSize
  is replaced by the hlo instruction the represents the dynamic size
  instead.

- Implements Dynamic Padder, which uses DynamicDimensionInference to
  analyze the graph and inserts certain operations to make sure the
  result is the same as if there is no padding in the bounded shapes.

PiperOrigin-RevId: 228266743
parent 09713e43
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