Searched refs:pool_mode (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | pooling.py | 146 functools.partial(backend.pool2d, pool_mode='max'),
188 functools.partial(backend.pool2d, pool_mode='avg'),
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 4658 pool_mode='max'): argument
4697 if pool_mode == 'max':
4700 elif pool_mode == 'avg':
4704 raise ValueError('Invalid pooling mode: ' + str(pool_mode))
4717 pool_mode='max'): argument
4750 if pool_mode == 'max':
4753 elif pool_mode == 'avg':
4757 raise ValueError('Invalid pooling mode: ' + str(pool_mode))
|
| D | backend_test.py | 687 pool_mode='max')
692 pool_mode='avg')
724 y = keras.backend.pool2d(x, (2, 2), strides=(2, 2), pool_mode='other')
731 pool_mode='max')
736 pool_mode='avg')
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.backend.pbtxt | 345 …argspec: "args=[\'x\', \'pool_size\', \'strides\', \'padding\', \'data_format\', \'pool_mode\'], v…
349 …argspec: "args=[\'x\', \'pool_size\', \'strides\', \'padding\', \'data_format\', \'pool_mode\'], v…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.backend.pbtxt | 349 …argspec: "args=[\'x\', \'pool_size\', \'strides\', \'padding\', \'data_format\', \'pool_mode\'], v…
353 …argspec: "args=[\'x\', \'pool_size\', \'strides\', \'padding\', \'data_format\', \'pool_mode\'], v…
|