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

Dynamic dimension inference

DynamicDimensionInference analyzes each HLO instruction in a graph and
inferences which dimensions are dynamic and which scalar instructions
represent the runtime real size of those dynamic dimensions.

Note: This CL only implements the skeleton that has enough support for
dynamic batch for Resnet50. More supports for other models will be
added incrementally later.
PiperOrigin-RevId: 224218787
parent d6bdfe7e
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment