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

  /external/tensorflow/tensorflow/compiler/tests/
pooling_ops_test.py 295 def _VerifyOneTest(self, pool_func, pool_grad_func, input_sizes, ksize,
301 pool_grad_func: Pooling gradient function for pool_grad_func
332 expected_input_gradients = pool_grad_func(
359 actual_input_gradients = pool_grad_func(
383 def _VerifyValues(self, pool_func, pool_grad_func, input_sizes, ksize,
389 pool_grad_func: Corresponding pooling gradient function.
396 self._VerifyOneTest(pool_func, pool_grad_func, input_sizes, ksize,
pooling_ops_3d_test.py 190 def _VerifyGradient(self, pool_func, pool_grad_func, input_sizes, ksize,
196 pool_grad_func: Pooling gradient function for pool_grad_func
227 expected_input_gradients = pool_grad_func(
242 actual_input_gradients = pool_grad_func(

Completed in 63 milliseconds