File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed
tensorflow/compiler/xla/service Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -15,6 +15,7 @@ limitations under the License.
1515
1616#include " tensorflow/compiler/xla/service/topk_rewriter.h"
1717
18+ #include < array>
1819#include < memory>
1920#include < optional>
2021#include < vector>
@@ -151,7 +152,7 @@ static bool IsNanSafeGt(HloComputation* comp) {
151152// Look for the instructions emitted from: xla/client/lib/sorting.cc
152153static bool HasIota (HloSortInstruction* sort, HloInstruction* data) {
153154 namespace m = match;
154- const auto sort_dims = {data->shape ().dimensions (sort->sort_dimension ())};
155+ const std::array< int64_t , 1 > sort_dims = {data->shape ().dimensions (sort->sort_dimension ())};
155156 auto match_iota = [](auto dims) {
156157 return m::Iota ().WithShape (m::Shape ().WithElementType (S32).WithDims (dims));
157158 };
You can’t perform that action at this time.
0 commit comments