Commit d81104a0 authored by Alina Sbirlea's avatar Alina Sbirlea Committed by TensorFlower Gardener
Browse files

Optimize dot(DynamicSlice(ConstA), ConstantB) by memoizing dot(ConstA, ConstB) 

Make transformation when ConstA and ConstB are 2D, and DynamicSlice is slicing a full row, column respectively.
Handle:
dot(DynamicSlice(Index, ConstA), ConstB) => DynamicSlice(Index, dot*(ConstA, ConstB));
and
dot(ConstA, DynamicSlice(Index, ConstB)) => DynamicSlice(Index, dot*(ConstA, ConstB));
PiperOrigin-RevId: 185891869
parent 972fa890
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