Home
last modified time | relevance | path

Searched refs:bias_ptr (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid_3x3_filter.h131 const int8* filter_ptr, const float* bias_ptr,
1029 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
1065 const int8* filter_ptr, const float* bias_ptr,
2035 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
2073 const int8* filter_ptr, const float* bias_ptr,
2167 [output_ptr] "+r"(output_ptr), [bias_ptr] "+r"(bias_ptr),
2189 const int8* filter_ptr, const float* bias_ptr,
2336 [output_ptr] "+r"(output_ptr), [bias_ptr] "+r"(bias_ptr),
2359 const int8* filter_ptr, const float* bias_ptr,
2542 [bias_ptr] "+r"(bias_ptr)
[all …]
Ddepthwise_conv_3x3_filter.h114 const int8* filter_ptr, const int32* bias_ptr,
932 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
969 const int8* filter_ptr, const int32* bias_ptr,
1893 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
1931 const int8* filter_ptr, const int32* bias_ptr,
2019 [output_ptr] "+r"(output_ptr), [bias_ptr] "+r"(bias_ptr),
2043 const int8* filter_ptr, const int32* bias_ptr,
2183 [output_ptr] "+r"(output_ptr), [bias_ptr] "+r"(bias_ptr),
2207 const int8* filter_ptr, const int32* bias_ptr,
2380 [output_ptr] "+r"(output_ptr), [bias_ptr] "+r"(bias_ptr),
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.h40 const float* recurrent_weights_ptr, const float* bias_ptr,
50 const float* recurrent_weights_ptr, const float* bias_ptr,
69 float recurrent_weights_scale, const float* bias_ptr, int input_size,
82 const float* bias_ptr, int input_size, int aux_input_size, int num_units,
Dkernel_utils.cc25 const float* recurrent_weights_ptr, const float* bias_ptr, in RnnBatchStep() argument
33 bias_ptr, input_size, /*aux_input_size=*/0, num_units, in RnnBatchStep()
41 const float* recurrent_weights_ptr, const float* bias_ptr, in RnnBatchStep() argument
50 tensor_utils::VectorBatchVectorAssign(bias_ptr, num_units, batch_size, in RnnBatchStep()
78 std::copy_n(bias_ptr, num_units, in RnnBatchStep()
122 float recurrent_weights_scale, const float* bias_ptr, int input_size, in RnnBatchStep() argument
133 recurrent_weights_scale, bias_ptr, input_size, in RnnBatchStep()
148 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep() argument
185 tensor_utils::VectorBatchVectorAssign(bias_ptr, num_units, batch_size, in RnnBatchStep()
255 std::copy_n(bias_ptr, num_units, in RnnBatchStep()
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc131 float* bias_ptr = static_cast<float*>(port::AlignedMalloc( in Compute() local
133 memset(bias_ptr, 0, bias_size * sizeof(float)); in Compute()
136 input_ptr, input_neon_dims, filter_ptr, filter_neon_dims, bias_ptr, in Compute()
140 port::AlignedFree(bias_ptr); in Compute()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_3x3_filter.h323 const int32* bias_ptr, uint8* output_ptr,
1220 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
1256 const int32* bias_ptr, uint8* output_ptr,
2045 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
2081 const int32* bias_ptr, uint8* output_ptr,
3077 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
3113 const int32* bias_ptr, uint8* output_ptr,
3999 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
4036 const int32* bias_ptr, uint8* output_ptr,
4128 [output_ptr] "+r"(output_ptr), [bias_ptr] "+r"(bias_ptr)
[all …]
Dlegacy_optimized_ops.h1179 const int32* bias_ptr = bias_data; in GEMVForLstmCell() local
1323 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCell()
1324 bias_ptr += 4; in GEMVForLstmCell()
1366 const int32* bias_ptr = bias_data; in GEMVForLstmCellWithSymmetricRange() local
1376 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCellWithSymmetricRange()
1377 bias_ptr += 4; in GEMVForLstmCellWithSymmetricRange()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsvdf.cc58 const float* const __restrict__ bias_ptr, TfLiteFusedActivation activation, in ApplyTimeWeightsBiasAndActivation() argument
77 if (bias_ptr) { in ApplyTimeWeightsBiasAndActivation()
81 const float* bias_data = bias_ptr; in ApplyTimeWeightsBiasAndActivation()
134 const float* bias_ptr = tflite::micro::GetTensorData<float>(bias); in EvalFloatSVDF() local
187 bias_ptr, params->activation, state_ptr, scratch_ptr, output_ptr); in EvalFloatSVDF()
Dsvdf_common.cc137 const int32_t* bias_ptr = bias_data; in EvalIntegerSvdfReference() local
139 *output_ptr++ = *bias_ptr++; in EvalIntegerSvdfReference()
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_rnn.cc214 const float* bias_ptr = GetTensorData<float>(bias); in EvalFloat() local
240 input_ptr_batch, input_weights_ptr, recurrent_weights_ptr, bias_ptr, in EvalFloat()
259 input_ptr_batch, input_weights_ptr, recurrent_weights_ptr, bias_ptr, in EvalFloat()
285 const float* bias_ptr = GetTensorData<float>(bias); in EvalHybrid() local
322 recurrent_weights_ptr, recurrent_weights_scale, bias_ptr, input_size, in EvalHybrid()
344 recurrent_weights_ptr, recurrent_weights_scale, bias_ptr, in EvalHybrid()
Dbasic_rnn.cc219 const float* bias_ptr = GetTensorData<float>(bias); in EvalFloat() local
222 input_ptr_batch, input_weights_ptr, recurrent_weights_ptr, bias_ptr, in EvalFloat()
253 const float* bias_ptr = GetTensorData<float>(bias); in EvalHybrid() local
271 recurrent_weights_ptr, recurrent_weights_scale, bias_ptr, input_size, in EvalHybrid()
/external/ruy/ruy/
Dkernel_avx512.cc678 const std::int32_t* bias_ptr = bias_col_ptr;
701 _mm512_loadu_si512(reinterpret_cast<const __m512i*>(bias_ptr));
702 bias_ptr += bias_ptr_block_increment;
915 const float* bias_ptr = params.bias;
948 bias_ptr + (col + 8 * mmm) * bias_ptr_block_increment;
959 _mm512_loadu_ps(bias_ptr + row * bias_ptr_block_increment);
1094 bias_ptr + (col + 8 * mmm) * bias_ptr_block_increment;
1105 _mm512_loadu_ps(bias_ptr + row * bias_ptr_block_increment);
1246 bias_ptr + (col + 8 * mmm) * bias_ptr_block_increment;
1257 _mm512_loadu_ps(bias_ptr + row * bias_ptr_block_increment);
[all …]
Dkernel_x86.h570 const float* bias_ptr = params.bias;
593 const float* bias_elem_ptr = bias_ptr + col * bias_ptr_block_increment;
598 const float* bias_elem_ptr = bias_ptr + row * bias_ptr_block_increment;
693 const float* bias_elem_ptr = bias_ptr + col * bias_ptr_block_increment;
698 const float* bias_elem_ptr = bias_ptr + row * bias_ptr_block_increment;
796 const float* bias_ptr = bias_col_ptr + row * bias_ptr_block_increment;
799 accum_data_v = _mm256_loadu_ps(bias_ptr);
848 const float* bias_ptr = bias_col_ptr + row * bias_ptr_block_increment;
851 accum_data_v = _mm256_loadu_ps(bias_ptr);
Dkernel_avx2_fma.cc755 const std::int32_t* bias_ptr = bias_col_ptr;
782 _mm256_loadu_si256(reinterpret_cast<const __m256i*>(bias_ptr));
783 bias_ptr += bias_ptr_block_increment;
Dkernel_avx.cc1196 const std::int32_t* bias_ptr = bias_col_ptr;
1223 _mm256_loadu_si256(reinterpret_cast<const __m256i*>(bias_ptr));
1224 bias_ptr += bias_ptr_block_increment;
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsvdf.cc74 const float* const __restrict__ bias_ptr, TfLiteFusedActivation activation, in ApplyTimeWeightsBiasAndActivation() argument
93 if (bias_ptr) { in ApplyTimeWeightsBiasAndActivation()
97 const float* bias_data = bias_ptr; in ApplyTimeWeightsBiasAndActivation()
150 const float* bias_ptr = tflite::micro::GetTensorData<float>(bias); in EvalFloatSVDF() local
203 bias_ptr, params->activation, state_ptr, scratch_ptr, output_ptr); in EvalFloatSVDF()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsvdf.h40 const float* const __restrict__ bias_ptr, TfLiteFusedActivation activation, in ApplyTimeWeightsBiasAndActivation() argument
56 if (bias_ptr) { in ApplyTimeWeightsBiasAndActivation()
57 tensor_utils::VectorBatchVectorAdd(bias_ptr, num_units, batch_size, in ApplyTimeWeightsBiasAndActivation()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsvdf.cc203 const int32_t* bias_ptr = bias_data; in EvalIntegerSvdfHifimini() local
205 *output_ptr++ = *bias_ptr++; in EvalIntegerSvdfHifimini()
300 const int32_t* bias_ptr = in EvalIntegerSvdfHifi4() local
308 output_ptr, vector1_ptr, vector2_ptr, bias_ptr, n_memory * n_rank, in EvalIntegerSvdfHifi4()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h636 se::DeviceMemory<T> bias_ptr = tensor_on_device(bias);
679 bias_desc, bias_ptr, // bias
692 output_ptr, bias_ptr, side_input_ptr, input_desc, filter_desc,