Home
last modified time | relevance | path

Searched refs:xnn_allocate_simd_memory (Results 1 – 13 of 13) sorted by relevance

/external/XNNPACK/src/xnnpack/
Dallocator.h50 inline static void* xnn_allocate_simd_memory(size_t memory_size) { in xnn_allocate_simd_memory() function
/external/XNNPACK/src/operators/
Dprelu-nc.c73 prelu_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_create_prelu_nc_f32()
Dsigmoid-nc.c115 sigmoid_op->lookup_table = xnn_allocate_simd_memory(256 * sizeof(uint8_t)); in xnn_create_sigmoid_nc_qu8()
Dleaky-relu-nc.c125 leaky_relu_op->lookup_table = xnn_allocate_simd_memory(256 * sizeof(uint8_t)); in xnn_create_leaky_relu_nc_qu8()
Dconvolution-nchw.c305 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_create_convolution2d_nchw_f32()
418 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_create_convolution2d_nchw_f32()
448 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_create_convolution2d_nchw_f32()
Dresize-bilinear-nhwc.c164 resize_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_setup_resize_bilinear2d_nhwc_f32()
Dresize-bilinear-nchw.c164 resize_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_setup_resize_bilinear2d_nchw_f32()
Dsoftmax-nc.c106 softmax_op->lookup_table = xnn_allocate_simd_memory(256 * sizeof(uint32_t)); in xnn_create_softmax_nc_qu8()
Dconvolution-nhwc.c243 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in create_convolution2d_nhwc()
268 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in create_convolution2d_nhwc()
312 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_group_weights_size * groups); in create_convolution2d_nhwc()
374 convolution_op->zero_buffer = xnn_allocate_simd_memory(zero_size); in create_convolution2d_nhwc()
Dfully-connected-nc.c106 fully_connected_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in create_fully_connected_nc()
Ddeconvolution-nhwc.c206 deconvolution_op->packed_weights = xnn_allocate_simd_memory(packed_group_weights_size * groups); in create_deconvolution2d_nhwc()
236 deconvolution_op->zero_buffer = xnn_allocate_simd_memory(zero_size); in create_deconvolution2d_nhwc()
Daverage-pooling-nhwc.c186 void* zero_buffer = xnn_allocate_simd_memory(zero_bytes); in xnn_create_average_pooling2d_nhwc_qu8()
/external/XNNPACK/src/
Druntime.c878 runtime->workspace = xnn_allocate_simd_memory(mem_arena_size); in xnn_create_runtime_v2()