experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Add AllGather to the HLO matchers.

Browse Source 
PiperOrigin-RevId: 312797129
Change-Id: I6a862f34a3b2331d99fe0bd242e21f26da7ed99e
This commit is contained in:
Yuanzhong Xu 2020-05-21 22:24:50 -07:00 committed by TensorFlower Gardener
parent dade83541f
commit e761103744
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/xla/service/hlo_matchers.h
View File

@ -203,6 +203,7 @@ HLO_MATCHER(Abs);
HLO_MATCHER(Add);
HLO_MATCHER(AddDependency);
HLO_MATCHER(AfterAll);
HLO_MATCHER(AllGather);
HLO_MATCHER(AllReduce);
HLO_MATCHER(AllToAll);
HLO_MATCHER(And);