| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | pool_test.py | 34 pooling_type, argument
77 if pooling_type == "MAX":
79 elif pooling_type == "AVG":
82 raise ValueError("Unsupported pooling type: %r" % (pooling_type,))
98 pooling_type, argument
135 pooling_type=pooling_type,
156 for pooling_type in ["MAX", "AVG"]:
161 pooling_type=pooling_type,
168 for pooling_type in ["MAX", "AVG"]:
177 pooling_type=pooling_type,
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | pooling_layer_builder.h | 30 TfLiteBuiltinOperator pooling_type) in PoolingLayerBuilder() argument
31 : OpBuilder(graph_builder), pooling_type_(pooling_type) {} in PoolingLayerBuilder()
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 186 def risc_pool(x, ksize, strides, pooling_type='MAX', name='RISC_POOL'): argument
188 x, ksize, strides, pooling_type=pooling_type, name=name)
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/kernels/ |
| D | a64_fp32_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::MAX; member
|
| D | sme_fp32_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::MAX; member
|
| D | a64_fp32_nhwc_avg_3x3_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::AVERAGE; member
|
| D | sme_fp32_nhwc_avg_3x3_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::AVERAGE; member
|
| D | sve_fp32_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 38 const static auto pooling_type = PoolingType::MAX; member
|
| D | sve_fp32_nhwc_avg_3x3_s1_output2x2_depthfirst.hpp | 38 const static auto pooling_type = PoolingType::AVERAGE; member
|
| D | sme_u8_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::MAX; member
|
| D | a64_s8_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::MAX; member
|
| D | a64_u8_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::MAX; member
|
| D | sme_s8_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::MAX; member
|
| D | sme_fp16_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::MAX; member
|
| D | sme_fp16_nhwc_avg_3x3_s1_output2x2_depthfirst.hpp | 36 const static auto pooling_type = PoolingType::AVERAGE; member
|
| D | sve_u8_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 38 const static auto pooling_type = PoolingType::MAX; member
|
| D | sve_s8_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 38 const static auto pooling_type = PoolingType::MAX; member
|
| D | a64_fp16_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 38 const static auto pooling_type = PoolingType::MAX; member
|
| D | sve_fp16_nhwc_max_2x2_s1_output2x2_depthfirst.hpp | 38 const static auto pooling_type = PoolingType::MAX; member
|
| D | sve_fp16_nhwc_avg_3x3_s1_output2x2_depthfirst.hpp | 38 const static auto pooling_type = PoolingType::AVERAGE; member
|
| D | a64_fp16_nhwc_avg_3x3_s1_output2x2_depthfirst.hpp | 38 const static auto pooling_type = PoolingType::AVERAGE; member
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | RiscPool.pbtxt | 24 name: "pooling_type"
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/ |
| D | pooling_implementation.hpp | 105 return ((args.pool_type == Strategy::pooling_type) && in is_supported()
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | freeze.py | 110 pooling_type='AVG',
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 1507 pooling_type, argument
1597 with ops.name_scope(name, "%s_pool" % (pooling_type.lower()),
1632 op_key = (pooling_type, num_spatial_dims)
1688 pooling_type, argument
1776 pooling_type=pooling_type,
|