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

Make sure configs are fully initialized in tests to fix msan warnings.

Browse Source 
PiperOrigin-RevId: 266135900
This commit is contained in:
A. Unique TensorFlower 2019-08-29 07:31:48 -07:00 committed by TensorFlower Gardener
parent 2820be1ef0
commit 5cbd7384d8
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/xla/tests/conv_depthwise_test.cc
View File

@ -60,6 +60,7 @@ static std::vector<DepthwiseConvolution2DSpec> GetConv2DTestCases() {
config.output_layout = {3, 0, 2, 1};
if (activation_size == 1 && kernel_size == 2) {
config.stride = config.pad = config.lhs_dilate = -1;
// Test for outer dim.
config.output_dims = {batch, activation_size + kernel_size - 1,
activation_size + kernel_size, feature};

1
tensorflow/compiler/xla/tests/grouped_convolution_test.cc
View File

@ -81,6 +81,7 @@ static std::vector<GroupedConvolution2DSpec> GetConv2DTestCases() {
config.kernel_layout = {3, 2, 1, 0};
if (activation_size == 1 && kernel_size == 2) {
config.stride = config.pad = config.lhs_dilate = -1;
// Test for outer dim.
config.output_dims = {batch, activation_size + kernel_size - 1,
activation_size + kernel_size, output_feature};