Home
last modified time | relevance | path

Searched refs:pool_func (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py108 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
139 input_sizes, total_size, pool_func, ksize, strides,
157 t = pool_func(
164 t = pool_func(
185 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
202 tf_logging.info("pool_func=%s", pool_func)
203 if pool_func == avg_pool_func:
207 self._VerifyOneType(pool_func, input_sizes, ksize, strides, padding,
209 self._VerifyOneType(pool_func, input_sizes, ksize, strides, padding,
213 self._VerifyOneType(pool_func, input_sizes, ksize, strides, padding,
[all …]
Dpooling_ops_3d_test.py48 def _VerifyOneTest(self, pool_func, input_sizes, window, strides, padding, argument
76 t = pool_func(
89 def _VerifyValues(self, pool_func, input_sizes, window, strides, argument
92 self._VerifyOneTest(pool_func, input_sizes, window, strides, padding,
209 pool_func, argument
240 if pool_func == nn_ops.avg_pool3d:
258 t = pool_func(
288 pool_func, argument
293 self._ConstructAndTestGradientForConfig(pool_func,
/external/tensorflow/tensorflow/compiler/tests/
Dpooling_ops_test.py74 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
100 t = pool_func(t,
110 def _VerifyValues(self, pool_func, input_sizes, ksize, strides, padding, argument
124 self._VerifyOneTest(pool_func, input_sizes, ksize, strides, padding,
296 pool_func, argument
332 outputs = pool_func(
447 pool_func, argument
467 pool_func,
Dpooling_ops_3d_test.py46 def _VerifyValues(self, pool_func, input_sizes, window, strides, padding, argument
67 t = pool_func(
191 pool_func, argument
218 outputs = pool_func(