Searched refs:pool_mode (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | pooling.py | 195 functools.partial(backend.pool2d, pool_mode='max'),
240 functools.partial(backend.pool2d, pool_mode='avg'),
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5785 pool_mode='max'): argument
5824 if pool_mode == 'max':
5827 elif pool_mode == 'avg':
5831 raise ValueError('Invalid pooling mode: ' + str(pool_mode))
5846 pool_mode='max'): argument
5879 if pool_mode == 'max':
5882 elif pool_mode == 'avg':
5886 raise ValueError('Invalid pooling mode: ' + str(pool_mode))
|
| D | backend_test.py | 804 pool_mode='max')
812 pool_mode='avg')
846 y = backend.pool2d(x, (2, 2), strides=(2, 2), pool_mode='other')
856 pool_mode='max')
864 pool_mode='avg')
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.backend.pbtxt | 353 …argspec: "args=[\'x\', \'pool_size\', \'strides\', \'padding\', \'data_format\', \'pool_mode\'], v…
357 …argspec: "args=[\'x\', \'pool_size\', \'strides\', \'padding\', \'data_format\', \'pool_mode\'], v…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.backend.pbtxt | 357 …argspec: "args=[\'x\', \'pool_size\', \'strides\', \'padding\', \'data_format\', \'pool_mode\'], v…
361 …argspec: "args=[\'x\', \'pool_size\', \'strides\', \'padding\', \'data_format\', \'pool_mode\'], v…
|