Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/
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()
/external/ComputeLibrary/src/cpu/kernels/
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
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()
/external/ComputeLibrary/tests/validation/reference/
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
/external/ComputeLibrary/src/core/NEON/kernels/
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
/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc220 const float* bias_ptr = GetTensorData<float>(bias); in EvalFloat() local
254 const float* bias_ptr = GetTensorData<float>(bias); in EvalHybrid() local
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
/external/ComputeLibrary/src/cpu/operators/
DCpuDepthwiseConv2dAssemblyDispatch.cpp121 …const auto bias_ptr = (bias) ? bias->buffer() + bias->info()->offset_first_element_in_bytes(… in prepare() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.cc118 se::DeviceMemory<T> output_ptr, se::DeviceMemory<T> bias_ptr, in AutotuneFusedConv()
Dmatmul_op_fused.cc264 auto bias_ptr = AsDeviceMemory(bias.template flat<T>().data(), in operator ()() local
/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
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()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsvdf.h40 const float* const __restrict__ bias_ptr, TfLiteFusedActivation activation, in ApplyTimeWeightsBiasAndActivation()
/external/ComputeLibrary/src/cpu/kernels/internal/
DCpuDepthwiseConv2dAssemblyWrapperKernel.cpp341 void CpuDepthwiseConv2dAssemblyWrapperKernel::pack_parameters(void *parameters_ptr, void *bias_ptr,… in pack_parameters()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1180 const int32* bias_ptr = bias_data; in GEMVForLstmCell() local
1367 const int32* bias_ptr = bias_data; in GEMVForLstmCellWithSymmetricRange() local