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

Fix minor bug in a convolution test.

Browse Source 
PiperOrigin-RevId: 277978950
Change-Id: I46d5454ae984de05f73290d81fa477278bd793bf
This commit is contained in:
A. Unique TensorFlower 2019-11-01 12:29:10 -07:00 committed by TensorFlower Gardener
parent bb2a75b548
commit 34b69b6c28
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/convolution_test.cc
View File

@ -1908,10 +1908,10 @@ XLA_TEST_F(ConvolutionTest, ConvolveF32BackwardInputGroupedConvolution) {
XlaBuilder builder(TestName());
Shape input_shape = ShapeUtil::MakeShape(F32, {1, 64, 100, 100});
Array4D<float> input_data(1, 64, 100, 100);
input_data.FillRandom(/*value=*/0.023, 0.001, /*seed=*/45321);
input_data.FillRandom(/*stddev=*/0.023, 0.001, /*seed=*/45321);
Shape filter_shape = ShapeUtil::MakeShape(F32, {7, 7, 1, 64});
Array4D<float> filter_data(7, 7, 1, 64);
input_data.FillRandom(/*value=*/0.023, 0.001, /*seed=*/45320);
filter_data.FillRandom(/*stddev=*/0.023, 0.001, /*seed=*/45320);
auto input = Parameter(&builder, 0, input_shape, "input");
auto filter = ConstantR4FromArray4D(&builder, filter_data);