Searched defs:pooling_type (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pool_test.py | 38 pooling_type, argument
102 pooling_type, argument
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | pooling_layer_builder.h | 28 TfLiteBuiltinOperator pooling_type) in PoolingLayerBuilder()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEPoolingLayerKernel.cpp | 669 …ling2_q8_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exc… in pooling2_q8_nchw()
796 …ing3_f16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exc… in pooling3_f16_nchw()
927 …ing2_f16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exc… in pooling2_f16_nchw()
999 …ling3_q8_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exc… in pooling3_q8_nchw()
1151 …gMxN_f16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exc… in poolingMxN_f16_nchw()
1391 …gMxN_f16_nhwc(const Window &window_input, const Window &window, PoolingType pooling_type, bool exc… in poolingMxN_f16_nhwc()
1561 …gMxN_f32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exc… in poolingMxN_f32_nchw()
1684 …rnel::pooling2_f32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, in pooling2_f32_nchw()
1754 …ing3_f32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exc… in pooling3_f32_nchw()
1821 …ing7_f32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, bool exc… in pooling7_f32_nchw()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 190 def risc_pool(x, ksize, strides, pooling_type='MAX', name='RISC_POOL'): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 1454 pooling_type, argument
1629 pooling_type, argument
|
| /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
|