diff options
| author |  Alina Sbirlea <asbirlea@google.com> | 2018-05-08 11:54:03 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-08 15:42:09 -0700 |
| commit | 59bffb7051231c7e0f8020892db8c3d584c555f4 (patch) | |
| tree | e485a0cfb02265a8eb04411874c9bd695183839e /configure.py | |
| parent | 4f7a0bc8c11827dde6986ad29e9fd21c48597367 (diff) | |
Re-land: 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));
Reason to roll forward: Previous issue of getting out of memory errors when generating LLVM constants was resolved by CSE-ing constants before allocation.
PiperOrigin-RevId: 195853680
Diffstat (limited to 'configure.py')
0 files changed, 0 insertions, 0 deletions