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

Increase maxpool test depth. Depth larger than kPoolingAccTrancheSize is needed to properly test the neon code.

Browse Source 
PiperOrigin-RevId: 240597128
This commit is contained in:
Jian Li 2019-03-27 10:55:01 -07:00 committed by TensorFlower Gardener
parent f366225ec6
commit ac88b10f3f
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/lite/kernels/internal/maxpool_quantized_test.cc
View File

@ -57,7 +57,7 @@ void RunOneMaxPoolTest(const PoolParams& params,
// test function. // test function.
void CreateDataAndRunMaxPool(bool padding_same) { void CreateDataAndRunMaxPool(bool padding_same) {
const int batch = UniformRandomInt(1, 20); const int batch = UniformRandomInt(1, 20);
const int input_depth = UniformRandomInt(32, 128); const int input_depth = UniformRandomInt(32, 700);
const int output_depth = input_depth; const int output_depth = input_depth;
const int input_width = UniformRandomInt(64, 128); const int input_width = UniformRandomInt(64, 128);
const int input_height = UniformRandomInt(64, 128); const int input_height = UniformRandomInt(64, 128);