Home
last modified time | relevance | path

Searched refs:pooling_type (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dpool_test.py38 pooling_type, argument
81 if pooling_type == "MAX":
83 elif pooling_type == "AVG":
86 raise ValueError("Unsupported pooling type: %r" % (pooling_type,))
102 pooling_type, argument
139 pooling_type=pooling_type,
160 for pooling_type in ["MAX", "AVG"]:
165 pooling_type=pooling_type,
172 for pooling_type in ["MAX", "AVG"]:
181 pooling_type=pooling_type,
[all …]
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning_utils.py166 pooling_type, argument
198 pooling_type=pooling_type,
206 pooling_type=pooling_type,
Dpruning.py469 pooling_type=self._block_pooling_function,
/external/tensorflow/tensorflow/examples/speech_commands/
Dfreeze.py110 pooling_type='AVG',
Dinput_data.py446 pooling_type='AVG',
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py1085 pooling_type, argument
1175 with ops.name_scope(name, "%s_pool" % (pooling_type.lower()),
1205 op_key = (pooling_type, num_spatial_dims)
1259 pooling_type, argument
1346 pooling_type=pooling_type,
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers_test.py202 output = _layers.pool(images, [3, 3], pooling_type='AVG')
210 images, [3, 3], pooling_type='AVG', data_format='NCHW')
218 images, [3, 3], pooling_type='AVG', outputs_collections='outputs')
226 output = _layers.pool(images, 3, pooling_type='AVG')
233 output = _layers.pool(images, [3, 3], pooling_type='MAX', scope='pool1')
239 output = _layers.pool(images, [3, 3], pooling_type='MAX', padding='SAME')
246 images, [3, 3], stride=1, padding='SAME', pooling_type='AVG')
253 images, images.get_shape()[1:3], stride=1, pooling_type='AVG')
259 output = _layers.pool(images, [2, 3], stride=[1, 2], pooling_type='AVG')
266 images, [2, 3], dilation_rate=[1, 2], pooling_type='AVG')
[all …]
Dlayers.py2469 pooling_type, argument
2514 with ops.name_scope(scope, '%s_pool' % (pooling_type.lower()),
2526 pooling_type=pooling_type,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt241 …argspec: "args=[\'input\', \'window_shape\', \'pooling_type\', \'strides\', \'padding\', \'data_fo…
/external/tensorflow/tensorflow/lite/python/
Dlite_test.py556 dconv, window_shape=[2, 2], pooling_type='AVG', padding='SAME')
558 pool, window_shape=[2, 2], pooling_type='MAX', padding='SAME')
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt293 …argspec: "args=[\'input\', \'window_shape\', \'pooling_type\', \'padding\', \'dilation_rate\', \'s…