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

Slight change to reduce_test to avoid generating inf, which was triggering an inf detector unnecessarily.

Browse Source 
PiperOrigin-RevId: 172965466
This commit is contained in:
Bjarke Hammersholt Roune 2017-10-20 19:44:11 -07:00 committed by TensorFlower Gardener
parent 93e8f3c67d
commit ba49d85832
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/tests/reduce_test.cc
View File

@ -457,7 +457,7 @@ XLA_TEST_F(ReduceTest, Reshape_111x2x25Reduce_111x50_To_R1) {
const Shape input_shape = ShapeUtil::MakeShape(F32, {rows, 2, cols / 2}); const Shape input_shape = ShapeUtil::MakeShape(F32, {rows, 2, cols / 2});
auto input = builder.Parameter(0, input_shape, "input"); auto input = builder.Parameter(0, input_shape, "input");
auto zero = builder.ConstantR0<float>(0.0); auto zero = builder.ConstantR0<float>(0.0);
auto log_ = builder.Log(input); auto log_ = builder.Tanh(input);
auto reshape = builder.Reshape(log_, {rows, cols}); auto reshape = builder.Reshape(log_, {rows, cols});
builder.Reduce(reshape, zero, add_f32, /*dimensions_to_reduce=*/{0}); builder.Reduce(reshape, zero, add_f32, /*dimensions_to_reduce=*/{0});
@ -473,7 +473,7 @@ XLA_TEST_F(ReduceTest, Reshape_111x2x25Reduce_111x50_To_R1) {
for (int64 colno = 0; colno < cols / 2; ++colno) { for (int64 colno = 0; colno < cols / 2; ++colno) {
float column_sum = 0; float column_sum = 0;
for (int64 rowno = 0; rowno < rows; ++rowno) { for (int64 rowno = 0; rowno < rows; ++rowno) {
column_sum += log(input_data(rowno, major, colno)); column_sum += tanh(input_data(rowno, major, colno));
} }
expected.push_back(column_sum); expected.push_back(column_sum);
} }