| /external/XNNPACK/src/ |
| D | convolution-nchw.c | 474 size_t input_batch_stride, in setup_convolution2d_nchw() argument
504 if (input_batch_stride < input_neurons) { in setup_convolution2d_nchw()
508 input_batch_stride, groups, group_input_channels, input_height, input_width); in setup_convolution2d_nchw()
584 .batched_a_stride = input_batch_stride << log2_input_element_size, in setup_convolution2d_nchw()
623 .input_batch_stride = input_batch_stride << log2_input_element_size, in setup_convolution2d_nchw()
663 .input_batch_stride = input_batch_stride << log2_input_element_size, in setup_convolution2d_nchw()
694 size_t input_batch_stride, in xnn_setup_convolution2d_nchw_f32() argument
709 batch_size, input_batch_stride, output_batch_stride, in xnn_setup_convolution2d_nchw_f32()
|
| D | operator-run.c | 229 (const void*) ((uintptr_t) context->input + batch_index * context->input_batch_stride), in xnn_compute_dconv2d_hwc2spchw()
263 …text->input + channel * context->input_channel_stride + batch_index * context->input_batch_stride), in xnn_compute_dwconv2d_spchw()
280 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_argmax_pooling_unipass()
300 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_argmax_pooling_multipass()
323 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_max_pooling()
437 (const void*) ((uintptr_t) context->input + batch_index * context->input_batch_stride); in xnn_compute_global_average_pooling_nwc_unipass()
456 (const void*) ((uintptr_t) context->input + batch_index * context->input_batch_stride); in xnn_compute_global_average_pooling_nwc_multipass()
479 channels_start * context->input_channel_stride + batch_index * context->input_batch_stride); in xnn_compute_global_average_pooling_ncw()
504 context->input_offset + batch_index * context->input_batch_stride, in xnn_compute_resize_bilinear()
|
| D | global-average-pooling-nwc.c | 287 .input_batch_stride = input_stride_in_bytes * width, in xnn_setup_global_average_pooling_nwc_q8()
351 .input_batch_stride = input_stride_in_bytes * width, in xnn_setup_global_average_pooling_nwc_f32()
|
| D | global-average-pooling-ncw.c | 132 .input_batch_stride = global_average_pooling_op->channels * width * sizeof(float), in xnn_setup_global_average_pooling_ncw_f32()
|
| D | resize-bilinear-nhwc.c | 192 .input_batch_stride = input_pixel_stride_in_bytes * input_height * input_width, in xnn_setup_resize_bilinear2d_nhwc_f32()
|
| D | argmax-pooling-nhwc.c | 256 ….input_batch_stride = input_height * input_width * argmax_pooling_op->input_pixel_stride * sizeof(… in xnn_setup_argmax_pooling2d_nhwc_f32()
|
| D | max-pooling-nhwc.c | 398 ….input_batch_stride = (input_height * input_width * max_pooling_op->input_pixel_stride) << log2_in… in setup_max_pooling2d()
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 213 size_t input_batch_stride; member
268 size_t input_batch_stride; member
297 size_t input_batch_stride; member
347 size_t input_batch_stride; member
456 size_t input_batch_stride; member
485 size_t input_batch_stride; member
511 size_t input_batch_stride; member
|
| /external/XNNPACK/test/ |
| D | convolution-nchw.cc | 222 .input_batch_stride(18013) in TEST()
688 .input_batch_stride(14879) in TEST()
928 .input_batch_stride(14879) in TEST()
1158 .input_batch_stride(14879) in TEST()
1398 .input_batch_stride(14879) in TEST()
|
| D | convolution-operator-tester.h | 341 inline ConvolutionOperatorTester& input_batch_stride(size_t input_batch_stride) { in input_batch_stride() function
342 assert(input_batch_stride >= 1); in input_batch_stride()
343 this->input_batch_stride_ = input_batch_stride; in input_batch_stride()
347 inline size_t input_batch_stride() const { in input_batch_stride() function
865 …batch_size() * input_batch_stride() + groups() * group_input_channels() * input_height() * input_w… in TestNCHWxF32()
941 input[i * input_batch_stride() + in TestNCHWxF32()
993 batch_size(), input_batch_stride(), output_batch_stride(), input_height(), input_width(), in TestNCHWxF32()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | depthwise_conv_hybrid.h | 140 const int input_batch_stride = input_height_stride * input_shape.Dims(1); in DepthwiseConvHybridGeneral() local
198 input_data + in_y * input_height_stride + b * input_batch_stride, in DepthwiseConvHybridGeneral()
|
| D | depthwise_conv.h | 1703 const int input_batch_stride = input_height_stride * input_shape.Dims(1);
1761 input_data + in_y * input_height_stride + b * input_batch_stride,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | depthwiseconv_float.h | 999 const int input_batch_stride = input_height_stride * input_shape.Dims(1);
1060 input_data + in_y * input_height_stride + b * input_batch_stride,
|
| D | depthwiseconv_uint8.h | 1777 const int input_batch_stride = input_height_stride * input_shape.Dims(1);
1837 input_data + in_y * input_height_stride + b * input_batch_stride,
|
| D | depthwiseconv_uint8_3x3_filter.h | 10252 const int input_batch_stride = input_height_stride * input_height;
10340 input_data_block = input_data + b * input_batch_stride +
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 905 size_t input_batch_stride,
|