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

Fix space_to_batch_converter on windows.

Browse Source 
PiperOrigin-RevId: 333398025
Change-Id: I4d713db2c910d462f1c70a1cd4979a9a3cfe0905
This commit is contained in:
Yujing Zhang 2020-09-23 16:33:37 -07:00 committed by TensorFlower Gardener
parent 7e49c60798
commit 055855171b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/xla/service/space_to_batch_converter.cc
View File

@ -371,8 +371,8 @@ Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) {
new_dim_numbers.set_output_feature_dimension(dim_count);
int p = 0;
for (auto [k, v] : dim_map) {
transpose_dims[p] = v;
for (const auto& entry : dim_map) {
transpose_dims[p] = entry.second;
p++;
}