Home
last modified time | relevance | path

Searched defs:bias_ptr (Results 1 – 25 of 28) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dkernel_utils.cc25 const float* recurrent_weights_ptr, const float* bias_ptr, in RnnBatchStep()
41 const float* recurrent_weights_ptr, const float* bias_ptr, in RnnBatchStep()
122 float recurrent_weights_scale, const float* bias_ptr, int input_size, in RnnBatchStep()
148 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuDirectConv2dOutputStageKernel.cpp194 const auto bias_ptr = reinterpret_cast<T *>(bi.ptr()) + x; in output_stage_nhwc() local
211 const auto bias_ptr = reinterpret_cast<T *>(bi.ptr()) + x; in output_stage_nhwc() local
352 const auto bias_ptr = reinterpret_cast<int32_t *>(bi.ptr()) + x; in output_stage_nhwc() local
374 const auto bias_ptr = reinterpret_cast<int32_t *>(bi.ptr()) + x; in output_stage_nhwc() local
H A DCpuGemmLowpOffsetContributionOutputStageKernel.cpp308 … int32_t *vector_sum_col_ptr, const int32_t *vector_sum_row_ptr, const int32_t *bias_ptr, Iterator… in run_offset_contribution_output_stage_window()
400 …tion_output_stage_window_symm(const int32_t *vector_sum_col_ptr, const int32_t *bias_ptr, Iterator… in run_offset_contribution_output_stage_window_symm()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DFullyConnectedLayer.cpp50 const TB *bias_ptr = bias.data(); in vector_matrix_multiply() local
68 const TB *bias_ptr = bias.data(); in vector_matrix_multiply() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEQLSTMLayerNormalizationKernel.cpp231 const int32_t *bias_ptr, in normalize_qasymm16()
295 const auto bias_ptr = reinterpret_cast<const BiasDataType *>(bias_iterator.ptr()); in compute_qsymm16() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbasic_rnn.cc220 const float* bias_ptr = GetTensorData<float>(bias); in EvalFloat() local
254 const float* bias_ptr = GetTensorData<float>(bias); in EvalHybrid() local
H A Dunidirectional_sequence_rnn.cc214 const float* bias_ptr = GetTensorData<float>(bias); in EvalFloat() local
285 const float* bias_ptr = GetTensorData<float>(bias); in EvalHybrid() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_convolution.cpp45 const char* const bias_ptr, in conv2d_impl()
287 const char* const bias_ptr = bias.has_value() in convolution_wrapper() local
H A Dop_linear_scratch_example.cpp111 scalar_t* bias_ptr = bias.value().mutable_data_ptr<scalar_t>() + j; in linear_scratch_example() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear_dynamic.cpp101 const float* bias_ptr = nullptr; in apply_dynamic_impl() local
273 const float* bias_ptr = bias_contig.const_data_ptr<float>(); in apply_dynamic_impl() local
H A Dqlinear.cpp103 const float* bias_ptr = nullptr; in apply_impl() local
343 const float* bias_ptr = nullptr; in apply_with_input_q_dq_qweight_dq_output_fp32_impl() local
H A Dqconv.cpp198 const float* PackedConvWeight<kSpatialDim>::GetBiasData(at::Tensor* bias_ptr) { in GetBiasData()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dgroup_norm_kernel.cpp235 const opmath_t* bias_ptr, in ApplyScaleBias()
254 const opmath_t* bias_ptr, in ApplyScaleBias()
362 opmath_t* bias_ptr = scale_ptr + D; in GroupNormKernelImplChannelsLastInternal() local
450 opmath_t* bias_ptr = scale_ptr + C; in GroupNormKernelImplChannelsLastInternal() local
476 opmath_t* bias_ptr = scale_ptr + C; in GroupNormKernelImplChannelsLastInternal() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
H A Dgemm_hybrid_indirect.hpp78 …ern_k, const Tro *b_ptr, size_t, IndirectOutputArg<Tr> output_arg, const Tr *bias_ptr, Activation … in run()
124 …nst Tro *b_ptr, size_t b_stride, IndirectOutputArg<Tr> output_arg, const Tr *bias_ptr, Activation … in run()
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuDepthwiseConv2dAssemblyDispatch.cpp121 …const auto bias_ptr = (bias) ? bias->buffer() + bias->info()->offset_first_element_in_bytes(… in prepare() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
H A Dqlinear.cpp84 const float* bias_ptr = nullptr; in apply_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu.cc118 se::DeviceMemory<T> output_ptr, se::DeviceMemory<T> bias_ptr, in AutotuneFusedConv()
H A Dmatmul_op_fused.cc264 auto bias_ptr = AsDeviceMemory(bias.template flat<T>().data(), in operator ()() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dsvdf.h40 const float* const __restrict__ bias_ptr, TfLiteFusedActivation activation, in ApplyTimeWeightsBiasAndActivation()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/internal/
H A DCpuDepthwiseConv2dAssemblyWrapperKernel.cpp341 void CpuDepthwiseConv2dAssemblyWrapperKernel::pack_parameters(void *parameters_ptr, void *bias_ptr,… in pack_parameters()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DLinear.cpp392 auto bias_ptr = bias_.data_ptr<float>(); in mkl_linear() local
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/
H A DGemmHipblaslt.h467 const void* bias_ptr = GetBiasPointerFromParams<CT>(params); in Call() local
H A DGemmCommon.h319 const void* bias_ptr; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDepthwiseConv3d.cu437 const scalar_t* bias_ptr = in conv_depthwise3d_cuda() local
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDABlas.cpp1405 const void* bias_ptr, in scaled_gemm()

12