Add sparse dot op support in algebraic simplifier (remove degenerate dimensions)
PiperOrigin-RevId: 613128127
Showing
- third_party/xla/xla/service/BUILD 1 addition, 0 deletionsthird_party/xla/xla/service/BUILD
- third_party/xla/xla/service/algebraic_simplifier.cc 57 additions, 16 deletionsthird_party/xla/xla/service/algebraic_simplifier.cc
- third_party/xla/xla/service/algebraic_simplifier.h 4 additions, 3 deletionsthird_party/xla/xla/service/algebraic_simplifier.h
- third_party/xla/xla/service/algebraic_simplifier_test.cc 36 additions, 0 deletionsthird_party/xla/xla/service/algebraic_simplifier_test.cc
- third_party/xla/xla/service/batch_dot_simplification.cc 7 additions, 0 deletionsthird_party/xla/xla/service/batch_dot_simplification.cc
- third_party/xla/xla/service/hlo_creation_utils.cc 7 additions, 6 deletionsthird_party/xla/xla/service/hlo_creation_utils.cc
- third_party/xla/xla/service/hlo_creation_utils.h 7 additions, 1 deletionthird_party/xla/xla/service/hlo_creation_utils.h
Please register or sign in to comment