Home
last modified time | relevance | path

Searched refs:input_batch_stride (Results 1 – 17 of 17) sorted by relevance

/external/XNNPACK/src/xnnpack/
Dcompute.h357 size_t input_batch_stride; member
387 size_t input_batch_stride; member
421 size_t input_batch_stride; member
477 size_t input_batch_stride; member
493 size_t input_batch_stride; member
543 size_t input_batch_stride; member
577 size_t input_batch_stride; member
613 size_t input_batch_stride; member
651 size_t input_batch_stride; member
682 size_t input_batch_stride; member
[all …]
/external/XNNPACK/src/
Doperator-run.c357 (const void*) ((uintptr_t) context->input + batch_index * context->input_batch_stride), in xnn_compute_conv2d_hwc2chw()
375 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_dwconv_unipass()
395 …text->input + channel * context->input_channel_stride + batch_index * context->input_batch_stride), in xnn_compute_dwconv2d_chw()
454 (const void*) ((uintptr_t) context->input + batch_index * context->input_batch_stride), in xnn_compute_depthtospace2d_chw2hwc()
466 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_argmax_pooling_unipass()
485 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_argmax_pooling_multipass()
507 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_max_pooling()
545 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_average_pooling_unipass()
563 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_average_pooling_multipass()
584 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_pixelwise_average_pooling_unipass()
[all …]
/external/XNNPACK/src/operators/
Dconvolution-nchw.c565 …const size_t input_batch_stride = (input_height * input_width * convolution_op->input_pixel_stride… in setup_convolution2d_nchw() local
604 .batched_input_stride = input_batch_stride, in setup_convolution2d_nchw()
645 .input_batch_stride = input_batch_stride, in setup_convolution2d_nchw()
698 .input_batch_stride = input_batch_stride, in setup_convolution2d_nchw()
Dglobal-average-pooling-ncw.c132 .input_batch_stride = global_average_pooling_op->channels * width * sizeof(float), in xnn_setup_global_average_pooling_ncw_f32()
Ddepth-to-space-nchw2nhwc.c139 ….input_batch_stride = depth_to_space_op->input_pixel_stride * input_height * input_width * sizeof(… in xnn_setup_depth_to_space_nchw2nhwc_x32()
Daverage-pooling-nhwc.c489 .input_batch_stride = input_stride_in_bytes * input_elements, in setup_average_pooling2d()
580 ….input_batch_stride = input_height * input_width * average_pooling_op->input_pixel_stride << log2_… in setup_average_pooling2d()
610 ….input_batch_stride = input_height * input_width * average_pooling_op->input_pixel_stride << log2_… in setup_average_pooling2d()
Dresize-bilinear-nhwc.c200 .input_batch_stride = input_pixel_stride_in_bytes * input_height * input_width, in xnn_setup_resize_bilinear2d_nhwc_f32()
Dresize-bilinear-nchw.c201 … .input_batch_stride = resize_op->input_pixel_stride * input_height * input_width * sizeof(float), in xnn_setup_resize_bilinear2d_nchw_f32()
Dargmax-pooling-nhwc.c263 ….input_batch_stride = input_height * input_width * argmax_pooling_op->input_pixel_stride * sizeof(… in xnn_setup_argmax_pooling2d_nhwc_f32()
Dmax-pooling-nhwc.c302 ….input_batch_stride = (input_height * input_width * max_pooling_op->input_pixel_stride) << log2_in… in setup_max_pooling2d_nhwc()
Dglobal-average-pooling-nwc.c178 .input_batch_stride = input_stride_in_bytes * width, in setup_global_average_pooling_nwc()
Dconvolution-nhwc.c1120 ….input_batch_stride = (input_height * input_width * convolution_op->input_pixel_stride) << log2_in… in setup_convolution2d_nhwc()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h140 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()
Ddepthwise_conv.h1711 const int input_batch_stride = input_height_stride * input_shape.Dims(1);
1769 input_data + in_y * input_height_stride + b * input_batch_stride,
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_float.h999 const int input_batch_stride = input_height_stride * input_shape.Dims(1);
1060 input_data + in_y * input_height_stride + b * input_batch_stride,
Ddepthwiseconv_uint8.h1777 const int input_batch_stride = input_height_stride * input_shape.Dims(1);
1837 input_data + in_y * input_height_stride + b * input_batch_stride,
Ddepthwiseconv_uint8_3x3_filter.h13230 const int input_batch_stride = input_height_stride * input_height;
13318 input_data_block = input_data + b * input_batch_stride +