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

Remove dependence on rand_r in xla/tests:reduce_test

Browse Source 
rand_r is not available on windows

PiperOrigin-RevId: 294813530
Change-Id: I4ff19ac7b5831b54d0825ebff81c2d3ec80b3e16
This commit is contained in:
Gunhan Gulsoy 2020-02-12 19:08:16 -08:00 committed by TensorFlower Gardener
parent f00437d811
commit 9afba0fc0d
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/compiler/xla/tests/reduce_test.cc
View File

@ -26,9 +26,11 @@ limitations under the License.
// <possible subsets of dimensions in chosen rank>
#include <stdlib.h>
#include <algorithm>
#include <cmath>
#include <memory>
#include <random>
#include <string>
#include <utility>
#include <vector>
@ -97,11 +99,12 @@ class ReduceTest : public ClientLibraryTestBase {
auto input = Parameter(&builder, 0, input_shape, "input");
auto zero = ConstantR0<float>(&builder, 0.0);
Reduce(input, zero, add_f32, /*dimensions_to_reduce=*/{0});
std::minstd_rand rng(seed_);
std::vector<float> input_data(element_count);
for (int64 i = 0; i < element_count; ++i) {
input_data[i] = rand_r(&seed_) % 3;
if (rand_r(&seed_) % 2 == 0) {
input_data[i] = rng() % 3;
if (rng() % 2 == 0) {
input_data[i] *= -1;
}
}