Home
last modified time | relevance | path

Searched defs:weights_ptr (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/depthwiseconv2d/generic/neon/
H A Dimpl.cpp180 auto weights_ptr = base_weights_ptr; in depthwise_loop_multiplier1_fp() local
214 auto weights_ptr = base_weights_ptr; in depthwise_loop_multiplier1_fp() local
287 auto weights_ptr = weights_it.ptr(); in depthwise_loop_generic_fp() local
387 auto weights_ptr = base_weights_ptr; in depthwise_loop_multiplier1_quantized() local
450 auto weights_ptr = base_weights_ptr; in depthwise_loop_multiplier1_quantized() local
556 auto weights_ptr = weights_it.ptr(); in depthwise_loop_generic_quantized() local
670 auto weights_ptr = weights_it.ptr(); in depthwise_loop_pow2_quantized_per_tensor() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DFullyConnectedLayer.cpp49 const T *weights_ptr = weights.data(); in vector_matrix_multiply() local
67 const T *weights_ptr = weights.data(); in vector_matrix_multiply() local
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dconvolution-nchw.c414 void* weights_ptr = xnn_get_pointer_to_write_weights( in xnn_create_convolution2d_nchw_f32() local
451 void* weights_ptr = xnn_get_pointer_to_write_weights( in xnn_create_convolution2d_nchw_f32() local
H A Dconvolution-nhwc.c402 void* weights_ptr = xnn_get_pointer_to_write_weights( in create_convolution2d_nhwc() local
435 void* weights_ptr = xnn_get_pointer_to_write_weights( in create_convolution2d_nhwc() local
501 void* weights_ptr = xnn_get_pointer_to_write_weights( in create_convolution2d_nhwc() local
H A Dprelu-nc.c94 void* weights_ptr = xnn_get_pointer_to_write_weights(prelu_op, aligned_total_weights_size, 0); in create_prelu_nc() local
H A Dfully-connected-nc.c123 void* weights_ptr = xnn_get_pointer_to_write_weights( in create_fully_connected_nc() local
H A Ddeconvolution-nhwc.c193 void* weights_ptr = xnn_get_pointer_to_write_weights( in create_deconvolution2d_nhwc() local
/aosp_15_r20/external/XNNPACK/src/
H A Doperator-utils.c17 void* weights_ptr = NULL; in xnn_get_pointer_to_write_weights() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuDepthwiseConv2dAssemblyDispatch.cpp120 … const auto weights_ptr = weights->buffer() + weights->info()->offset_first_element_in_bytes(); in prepare() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/
H A Dmp8x27-sse2.c41 } weights_ptr = {weights}; in pytorch_q8dwconv_ukernel_mp8x27__sse2() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dembedding_lookup_sparse.cc203 const float* weights_ptr = GetTensorData<float>(weights); in Eval() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/internal/
H A DCpuDepthwiseConv2dAssemblyWrapperKernel.cpp341 …rapperKernel::pack_parameters(void *parameters_ptr, void *bias_ptr, void *weights_ptr, size_t ld_w… in pack_parameters()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc1961 const auto* weights_ptr = in ConvertConv2DHelper() local
2090 const int* weights_ptr = weights.GetPointer<int>(); in ConvertTranspose() local
5644 const int* weights_ptr = inputs.at(1).weights().GetPointer<int>(); in ConvertResize() local
5668 const int* weights_ptr = inputs.at(1).weights().GetPointer<int>(); in ConvertResize() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dlegacy_optimized_ops.h1193 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell() local
1269 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell() local