| /external/XNNPACK/test/ |
| D | x32-pad.cc | 59 .pre_padding(4) in TEST()
65 for (size_t pre_padding = 8; pre_padding < 32; pre_padding += 4) { in TEST() local
69 .pre_padding(pre_padding) in TEST()
76 for (size_t pre_padding = 1; pre_padding < 4; pre_padding++) { in TEST() local
80 .pre_padding(pre_padding) in TEST()
87 for (size_t pre_padding = 5; pre_padding < 8; pre_padding++) { in TEST() local
91 .pre_padding(pre_padding) in TEST()
133 .pre_padding(post_padding) in TEST()
145 .pre_padding(channels) in TEST()
159 .pre_padding(channels) in TEST()
[all …]
|
| D | pad-microkernel-tester.h | 44 inline PadMicrokernelTester& pre_padding(size_t pre_padding) { in pre_padding() function
45 this->pre_padding_ = pre_padding; in pre_padding()
49 inline size_t pre_padding() const { in pre_padding() function
63 return pre_padding() + input_channels() + post_padding(); in output_channels()
89 return pre_padding() + input_channels() + post_padding(); in output_stride()
91 assert(this->output_stride_ >= pre_padding() + input_channels() + post_padding()); in output_stride()
111 …std::vector<uint32_t> output((pre_padding() + input_channels() + post_padding()) + (rows() - 1) * … in Test()
112 std::vector<uint32_t> output_ref(rows() * (pre_padding() + input_channels() + post_padding())); in Test()
124 &output_ref[i * output_channels() + pre_padding()]); in Test()
131 pre_padding() * sizeof(uint32_t), in Test()
[all …]
|
| D | constant-pad-operator-tester.h | 58 inline size_t pre_padding(size_t i) const { in pre_padding() function
85 return pre_padding(i) + input_dim(i) + post_padding(i); in output_dim()
124 input_pre_paddings[XNN_MAX_TENSOR_DIMS - num_dims() + i] = pre_padding(i); in TestX32()
|
| /external/XNNPACK/src/x32-pad/ |
| D | scalar-float.c | 14 size_t pre_padding, in xnn_x32_pad_ukernel__scalar_float() argument
23 assert(pre_padding % sizeof(uint32_t) == 0); in xnn_x32_pad_ukernel__scalar_float()
28 const size_t output_increment = output_stride - (pre_padding + channels + post_padding); in xnn_x32_pad_ukernel__scalar_float()
35 size_t l = pre_padding; in xnn_x32_pad_ukernel__scalar_float()
|
| D | scalar-int.c | 14 size_t pre_padding, in xnn_x32_pad_ukernel__scalar_int() argument
23 assert(pre_padding % sizeof(uint32_t) == 0); in xnn_x32_pad_ukernel__scalar_int()
28 const size_t output_increment = output_stride - (pre_padding + channels + post_padding); in xnn_x32_pad_ukernel__scalar_int()
33 size_t l = pre_padding; in xnn_x32_pad_ukernel__scalar_int()
|
| D | wasmsimd.c | 16 size_t pre_padding, in xnn_x32_pad_ukernel__wasmsimd() argument
25 assert(pre_padding % sizeof(uint32_t) == 0); in xnn_x32_pad_ukernel__wasmsimd()
30 const size_t output_increment = output_stride - (pre_padding + channels + post_padding); in xnn_x32_pad_ukernel__wasmsimd()
35 size_t l = pre_padding; in xnn_x32_pad_ukernel__wasmsimd()
|
| D | sse.c | 16 size_t pre_padding, in xnn_x32_pad_ukernel__sse() argument
25 assert(pre_padding % sizeof(uint32_t) == 0); in xnn_x32_pad_ukernel__sse()
30 const size_t output_increment = output_stride - (pre_padding + channels + post_padding); in xnn_x32_pad_ukernel__sse()
37 size_t l = pre_padding; in xnn_x32_pad_ukernel__sse()
|
| D | neon.c | 16 size_t pre_padding, in xnn_x32_pad_ukernel__neon() argument
25 assert(pre_padding % sizeof(uint32_t) == 0); in xnn_x32_pad_ukernel__neon()
30 const size_t output_increment = output_stride - (pre_padding + channels + post_padding); in xnn_x32_pad_ukernel__neon()
35 size_t l = pre_padding; in xnn_x32_pad_ukernel__neon()
|
| /external/XNNPACK/src/operators/ |
| D | constant-pad-nd.c | 123 const size_t pre_padding = pre_paddings[num_dims - 1 - i]; in setup_constant_pad_nd() local
127 const bool is_current_dim_padded = (pre_padding | post_padding) != 0; in setup_constant_pad_nd()
129 normalized_pre_paddings[XNN_MAX_TENSOR_DIMS - 1 - num_squeezed_dims] = pre_padding; in setup_constant_pad_nd()
131 …normalized_output_shape[XNN_MAX_TENSOR_DIMS - 1 - num_squeezed_dims] = pre_padding + input_dim + p… in setup_constant_pad_nd()
137 assert(pre_padding == 0); in setup_constant_pad_nd()
190 const size_t* pre_padding, in xnn_setup_constant_pad_nd_x32() argument
198 num_dims, input_shape, pre_padding, post_padding, in xnn_setup_constant_pad_nd_x32()
|
| /external/XNNPACK/src/xnnpack/ |
| D | pad.h | 23 size_t pre_padding, \
|
| D | params.h | 952 size_t pre_padding,
963 size_t pre_padding,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2457 nvinfer1::DimsHW pre_padding(0, 0); in ConvertConv2DHelper() local
2460 params->converter->network()->addPadding(*output_tensor, pre_padding, in ConvertConv2DHelper()
2973 nvinfer1::DimsHW pre_padding, post_padding; in ConvertStridedSliceHelper() local
2976 pre_padding.d[i] = -begin[axis]; in ConvertStridedSliceHelper()
3009 *tensor, pre_padding, post_padding); in ConvertStridedSliceHelper()
4900 nvinfer1::DimsHW pre_padding(0, 0); in ConvertPad() local
4932 pre_padding.d[k] = pad_data[tf_index * 2]; in ConvertPad()
4946 *tensor, pre_padding, post_padding); in ConvertPad()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 2402 const int32_t pre_padding = paddings_data[i * 2 + 0]; in VisitPadNode() local
2403 if (pre_padding < 0) { in VisitPadNode()
2406 "invalid pre-padding %d for dimension #%d in node %d", pre_padding, in VisitPadNode()
2415 "invalid post-padding %d for dimension #%d in node %d", pre_padding, in VisitPadNode()
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 1724 const size_t* pre_padding,
|