HomeSort by relevance Sort by last modified time
    Searched refs:pool_func (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/python/kernel_tests/
pooling_ops_test.py 95 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding,
100 pool_func: Function to be called, co.MaxPool, co.AvgPool,
126 input_sizes, total_size, pool_func, ksize, strides)
143 t = pool_func(
150 t = pool_func(
171 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding,
176 pool_func: Function to be called, co.MaxPool, co.AvgPool,
188 tf_logging.info("pool_func=%s", pool_func)
189 if pool_func == avg_pool_func
    [all...]
pooling_ops_3d_test.py 48 def _VerifyOneTest(self, pool_func, input_sizes, window, strides, padding,
53 pool_func: Function to be called: co.MaxPool, co.AvgPool.
76 t = pool_func(
89 def _VerifyValues(self, pool_func, input_sizes, window, strides,
92 self._VerifyOneTest(pool_func, input_sizes, window, strides, padding,
209 pool_func,
220 pool_func: Function to be called, co.MaxPool, co.AvgPool,
240 if pool_func == nn_ops.avg_pool3d:
257 t = pool_func(
287 pool_func,
    [all...]
  /external/tensorflow/tensorflow/compiler/tests/
pooling_ops_test.py 74 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding,
79 pool_func: Function to be called, currently only co.MaxPool.
100 t = pool_func(t,
110 def _VerifyValues(self, pool_func, input_sizes, ksize, strides, padding,
115 pool_func: Function to be called, co.MaxPool, co.AvgPool,
124 self._VerifyOneTest(pool_func, input_sizes, ksize, strides, padding,
295 def _VerifyOneTest(self, pool_func, pool_grad_func, input_sizes, ksize,
300 pool_func: Forward pooling function
315 outputs = pool_func(
383 def _VerifyValues(self, pool_func, pool_grad_func, input_sizes, ksize
    [all...]
pooling_ops_3d_test.py 46 def _VerifyValues(self, pool_func, input_sizes, window, strides, padding,
51 pool_func: Function to be called: co.MaxPool, co.AvgPool.
67 t = pool_func(
190 def _VerifyGradient(self, pool_func, pool_grad_func, input_sizes, ksize,
195 pool_func: Forward pooling function
211 outputs = pool_func(

Completed in 367 milliseconds