Home
last modified time | relevance | path

Searched refs:XNN_FLAG_TENSORFLOW_SAME_PADDING (Results 1 – 16 of 16) sorted by relevance

/external/XNNPACK/src/subgraph/
Ddepthwise-convolution-2d.c100 const uint32_t supported_flags = XNN_FLAG_TENSORFLOW_SAME_PADDING; in xnn_define_depthwise_convolution_2d()
110 if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0 && any_padding) { in xnn_define_depthwise_convolution_2d()
120 …if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0 && (subsampling_height | subsampling_width) ==… in xnn_define_depthwise_convolution_2d()
121 flags &= ~XNN_FLAG_TENSORFLOW_SAME_PADDING; in xnn_define_depthwise_convolution_2d()
Dconvolution-2d.c108 const uint32_t supported_flags = XNN_FLAG_TENSORFLOW_SAME_PADDING; in xnn_define_convolution_2d()
118 if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0 && any_padding) { in xnn_define_convolution_2d()
128 …if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0 && (subsampling_height | subsampling_width) ==… in xnn_define_convolution_2d()
129 flags &= ~XNN_FLAG_TENSORFLOW_SAME_PADDING; in xnn_define_convolution_2d()
Daverage-pooling-2d.c84 if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0) { in xnn_define_average_pooling_2d()
Dmax-pooling-2d.c92 if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0) { in xnn_define_max_pooling_2d()
/external/XNNPACK/src/operators/
Dargmax-pooling-nhwc.c106 if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0) { in xnn_create_argmax_pooling2d_nhwc_f32()
199 if (argmax_pooling_op->flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) { in xnn_setup_argmax_pooling2d_nhwc_f32()
Daverage-pooling-nhwc.c143 if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0) { in xnn_create_average_pooling2d_nhwc_qu8()
335 if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0) { in xnn_create_average_pooling2d_nhwc_f32()
384 const bool tf_same_padding = (flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0; in xnn_create_average_pooling2d_nhwc_f32()
447 const bool tf_same_padding = (average_pooling_op->flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0; in setup_average_pooling2d()
Dmax-pooling-nhwc.c141 if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0) { in create_max_pooling2d_nhwc()
230 if (max_pooling_op->flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) { in setup_max_pooling2d_nhwc()
Ddeconvolution-nhwc.c149 if (any_padding && (flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0) { in create_deconvolution2d_nhwc()
273 if (flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) { in create_deconvolution2d_nhwc()
287 deconvolution_op->flags = XNN_FLAG_TENSORFLOW_SAME_PADDING; in create_deconvolution2d_nhwc()
825 if (deconvolution_op->flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) { in setup_deconvolution2d_nhwc()
Dconvolution-nhwc.c196 if ((flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0) { in create_convolution2d_nhwc()
372 const bool tf_same_padding = (flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) != 0 && kernel_size != 1; in create_convolution2d_nhwc()
405 convolution_op->flags |= XNN_FLAG_TENSORFLOW_SAME_PADDING; in create_convolution2d_nhwc()
811 if (convolution_op->flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) { in setup_convolution2d_nhwc()
/external/XNNPACK/src/
Dsubgraph.c176 if (node->flags & XNN_FLAG_TENSORFLOW_SAME_PADDING) { in xnn_check_nchw_compatibility()
607 … if (is_spatial_2d_zero_padding && !(consumer->flags & XNN_FLAG_TENSORFLOW_SAME_PADDING)) { in xnn_subgraph_optimize()
631 … if (is_spatial_2d_zero_padding && !(consumer->flags & XNN_FLAG_TENSORFLOW_SAME_PADDING)) { in xnn_subgraph_optimize()
/external/XNNPACK/test/
Dmax-pooling-operator-tester.h499 padding_tf_same() ? XNN_FLAG_TENSORFLOW_SAME_PADDING : 0, in TestU8()
596 padding_tf_same() ? XNN_FLAG_TENSORFLOW_SAME_PADDING : 0, in TestF32()
Dargmax-pooling-operator-tester.h385 padding_tf_same() ? XNN_FLAG_TENSORFLOW_SAME_PADDING : 0, in TestF32()
Dconvolution-operator-tester.h633 …? XNN_FLAG_DEPTHWISE_CONVOLUTION : 0) | (padding_tf_same() ? XNN_FLAG_TENSORFLOW_SAME_PADDING : 0), in TestNHWCxQS8()
802 …? XNN_FLAG_DEPTHWISE_CONVOLUTION : 0) | (padding_tf_same() ? XNN_FLAG_TENSORFLOW_SAME_PADDING : 0), in TestNHWCxQU8()
961 …? XNN_FLAG_DEPTHWISE_CONVOLUTION : 0) | (padding_tf_same() ? XNN_FLAG_TENSORFLOW_SAME_PADDING : 0), in TestNHWCxF32()
1123 …? XNN_FLAG_DEPTHWISE_CONVOLUTION : 0) | (padding_tf_same() ? XNN_FLAG_TENSORFLOW_SAME_PADDING : 0), in TestNHWCxF16()
Ddeconvolution-operator-tester.h586 padding_tf_same() ? XNN_FLAG_TENSORFLOW_SAME_PADDING : 0, in TestQU8()
719 padding_tf_same() ? XNN_FLAG_TENSORFLOW_SAME_PADDING : 0, in TestF32()
/external/XNNPACK/include/
Dxnnpack.h45 #define XNN_FLAG_TENSORFLOW_SAME_PADDING 0x00000004 macro
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc324 *flags = XNN_FLAG_TENSORFLOW_SAME_PADDING; in CalculatePadding()