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

Fix bad index type for Eigen::DSizes.

Browse Source 
PiperOrigin-RevId: 275398984
Change-Id: I854e234e34ac26734d385353d1e3999d2665547d
This commit is contained in:
A. Unique TensorFlower 2019-10-17 21:10:11 -07:00 committed by TensorFlower Gardener
parent 1d74adf892
commit be536832d9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/lrn_op_test.cc
View File

@ -67,7 +67,7 @@ class LRNFloatTest : public OpsTestBase {
Eigen::Tensor<float, 4, Eigen::RowMajor> expected(batch_size, rows, cols, Eigen::Tensor<float, 4, Eigen::RowMajor> expected(batch_size, rows, cols,
depth); depth);
auto out = expected.reshape(Eigen::DSizes<int64, 2>{rest, depth}); auto out = expected.reshape(Eigen::DSizes<Eigen::Index, 2>{rest, depth});
auto in = input.shaped<float, 2>({rest, depth}); auto in = input.shaped<float, 2>({rest, depth});
for (int64 i = 0; i < rest; ++i) { for (int64 i = 0; i < rest; ++i) {