Searched refs:log2_element_size (Results 1 – 6 of 6) sorted by relevance
| /external/XNNPACK/src/xnnpack/ |
| D | indirection.h | 25 uint32_t log2_element_size);
31 uint32_t log2_element_size);
36 uint32_t log2_element_size);
41 uint32_t log2_element_size);
47 uint32_t log2_element_size);
76 uint32_t log2_element_size);
|
| /external/XNNPACK/src/operators/ |
| D | channel-shuffle-nc.c | 142 uint32_t log2_element_size, in setup_channel_shuffle_nc() argument
165 .x_stride = channel_shuffle_op->input_pixel_stride << log2_element_size, in setup_channel_shuffle_nc()
167 .y_stride = channel_shuffle_op->output_pixel_stride << log2_element_size, in setup_channel_shuffle_nc()
168 .n = channel_shuffle_op->group_channels << log2_element_size, in setup_channel_shuffle_nc()
|
| D | unary-elementwise-nc.c | 91 uint32_t log2_element_size, in setup_unary_elementwise_nc() argument
116 .x_stride = input_stride << log2_element_size, in setup_unary_elementwise_nc()
118 .y_stride = output_stride << log2_element_size, in setup_unary_elementwise_nc()
126 unary_elementwise_op->compute.range[0] = (batch_size * channels) << log2_element_size; in setup_unary_elementwise_nc()
130 .n = channels << log2_element_size, in setup_unary_elementwise_nc()
132 .x_stride = input_stride << log2_element_size, in setup_unary_elementwise_nc()
134 .y_stride = output_stride << log2_element_size, in setup_unary_elementwise_nc()
|
| D | global-average-pooling-nwc.c | 30 uint32_t log2_element_size, in create_global_average_pooling_nwc() argument
90 const size_t zero_size = (channels << log2_element_size) + XNN_EXTRA_BYTES; in create_global_average_pooling_nwc()
123 size_t log2_element_size, in setup_global_average_pooling_nwc() argument
172 … size_t input_stride_in_bytes = global_average_pooling_op->input_pixel_stride << log2_element_size; in setup_global_average_pooling_nwc()
182 .output_batch_stride = (global_average_pooling_op->output_pixel_stride << log2_element_size), in setup_global_average_pooling_nwc()
|
| D | binary-elementwise-nd.c | 383 uint32_t log2_element_size, in setup_binary_elementwise_nd() argument
519 .elements = compressed_output_shape[0] << log2_element_size, in setup_binary_elementwise_nd()
544 …->context.elementwise_binary.a_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = a_stride << log2_element_size; in setup_binary_elementwise_nd()
547 …->context.elementwise_binary.b_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = b_stride << log2_element_size; in setup_binary_elementwise_nd()
549 …->context.elementwise_binary.y_stride[XNN_MAX_TENSOR_DIMS - 1 - i] = y_stride << log2_element_size; in setup_binary_elementwise_nd()
|
| /external/XNNPACK/src/ |
| D | indirection.c | 21 uint32_t log2_element_size) in xnn_indirection_init_conv2d() argument
26 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_conv2d()
81 uint32_t log2_element_size) in xnn_indirection_init_deconv2d() argument
85 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_deconv2d()
136 uint32_t log2_element_size) in xnn_indirection_init_subconv2d() argument
141 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_subconv2d()
202 uint32_t log2_element_size) in xnn_indirection_init_dwconv2d() argument
206 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_dwconv2d()
253 uint32_t log2_element_size) in xnn_indirection_init_maxpool2d() argument
257 const size_t input_pixel_stride = op->input_pixel_stride << log2_element_size; in xnn_indirection_init_maxpool2d()
[all …]
|