Searched refs:pooling_type (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pool_test.py | 38 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/lite/delegates/coreml/builders/ |
| D | pooling_layer_builder.h | 28 TfLiteBuiltinOperator pooling_type) in PoolingLayerBuilder() argument
29 : OpBuilder(graph_builder), pooling_type_(pooling_type) {} in PoolingLayerBuilder()
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 190 def risc_pool(x, ksize, strides, pooling_type='MAX', name='RISC_POOL'): argument
192 x, ksize, strides, pooling_type=pooling_type, name=name)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | RiscPool.pbtxt | 24 name: "pooling_type"
|
| /external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/ |
| D | pooling.cc | 176 const MliPoolingType pooling_type) { in EvalMli() argument
225 if (pooling_type == AveragePooling) in EvalMli()
227 else if (pooling_type == MaxPooling) in EvalMli()
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | freeze.py | 114 pooling_type='AVG',
|
| D | input_data.py | 449 pooling_type='AVG',
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 1334 pooling_type, argument
1424 with ops.name_scope(name, "%s_pool" % (pooling_type.lower()),
1454 op_key = (pooling_type, num_spatial_dims)
1509 pooling_type, argument
1596 pooling_type=pooling_type,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 249 …argspec: "args=[\'input\', \'window_shape\', \'pooling_type\', \'strides\', \'padding\', \'data_fo…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 293 …argspec: "args=[\'input\', \'window_shape\', \'pooling_type\', \'padding\', \'dilation_rate\', \'s…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 42673 name: "pooling_type"
|