| /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/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEPoolingLayerKernel.h | 88 …32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
102 …32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
110 …32_nhwc(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
118 …32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
126 …32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
134 …16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
155 …16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
163 …16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
171 …16_nhwc(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
180 …q8_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p…
[all …]
|
| D | NEPoolingLayerKernel.cpp | 669 …q8_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p… in pooling2_q8_nchw() argument
714 if(pooling_type != PoolingType::MAX) in pooling2_q8_nchw()
796 …16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p… in pooling3_f16_nchw() argument
798 ARM_COMPUTE_UNUSED(pooling_type); in pooling3_f16_nchw()
827 if(pooling_type == PoolingType::L2) in pooling3_f16_nchw()
834 if(pooling_type != PoolingType::MAX) in pooling3_f16_nchw()
852 if(pooling_type == PoolingType::L2) in pooling3_f16_nchw()
927 …16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exclude_p… in pooling2_f16_nchw() argument
929 ARM_COMPUTE_UNUSED(pooling_type); in pooling2_f16_nchw()
932 if(pooling_type == PoolingType::MAX && _indices) in pooling2_f16_nchw()
[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/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 | 1454 pooling_type, argument
1544 with ops.name_scope(name, "%s_pool" % (pooling_type.lower()),
1574 op_key = (pooling_type, num_spatial_dims)
1629 pooling_type, argument
1716 pooling_type=pooling_type,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 253 …argspec: "args=[\'input\', \'window_shape\', \'pooling_type\', \'strides\', \'padding\', \'data_fo…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 297 …argspec: "args=[\'input\', \'window_shape\', \'pooling_type\', \'padding\', \'dilation_rate\', \'s…
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 2431 PoolingType pooling_type = getPoolingType(op, result.getResultNumber()); in matchAndRewrite() local
2432 switch (pooling_type) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 42085 name: "pooling_type"
|