Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc23 const float* recurrent_weights_ptr, const float* bias_ptr, in RnnBatchStep() argument
31 bias_ptr, input_size, /*aux_input_size=*/0, num_units, in RnnBatchStep()
39 const float* recurrent_weights_ptr, const float* bias_ptr, in RnnBatchStep() argument
48 tensor_utils::VectorBatchVectorAssign(bias_ptr, num_units, batch_size, in RnnBatchStep()
76 tensor_utils::CopyVector(bias_ptr, num_units, in RnnBatchStep()
123 float recurrent_weights_scale, const float* bias_ptr, int input_size, in RnnBatchStep() argument
132 recurrent_weights_scale, bias_ptr, input_size, in RnnBatchStep()
145 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep() argument
155 tensor_utils::VectorBatchVectorAssign(bias_ptr, num_units, batch_size, in RnnBatchStep()
229 tensor_utils::CopyVector(bias_ptr, num_units, in RnnBatchStep()
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,
80 const float* bias_ptr, int input_size, int aux_input_size, int num_units,
/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/
Dunidirectional_sequence_rnn.cc153 const float* bias_ptr = bias->data.f; in EvalFloat() local
178 input_ptr_batch, input_weights_ptr, recurrent_weights_ptr, bias_ptr, in EvalFloat()
195 input_ptr_batch, input_weights_ptr, recurrent_weights_ptr, bias_ptr, in EvalFloat()
219 const float* bias_ptr = bias->data.f; in EvalHybrid() local
259 recurrent_weights_ptr, recurrent_weights_scale, bias_ptr, input_size, in EvalHybrid()
278 recurrent_weights_ptr, recurrent_weights_scale, bias_ptr, in EvalHybrid()
Dbasic_rnn.cc165 const float* bias_ptr = bias->data.f; in EvalFloat() local
168 input_ptr_batch, input_weights_ptr, recurrent_weights_ptr, bias_ptr, in EvalFloat()
199 const float* bias_ptr = bias->data.f; in EvalHybrid() local
211 recurrent_weights_ptr, recurrent_weights_scale, bias_ptr, input_size, in EvalHybrid()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_3x3_filter.h422 const int32* bias_ptr, uint8* output_ptr,
1371 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
1406 const int32* bias_ptr, uint8* output_ptr,
2454 [bias_ptr] "r"(bias_ptr), [input_row_size] "r"(input_row_size),
2495 const int32* bias_ptr, uint8* output_ptr,
2594 [output_ptr] "+r"(output_ptr), [bias_ptr] "+r"(bias_ptr)
2614 const int32* bias_ptr, uint8* output_ptr,
2765 [output_ptr] "+r"(output_ptr), [bias_ptr] "+r"(bias_ptr)
2785 const int32* bias_ptr, uint8* output_ptr,
2969 [output_ptr] "+r"(output_ptr), [bias_ptr] "+r"(bias_ptr)
[all …]
Doptimized_ops.h318 const int32* bias_ptr = bias_data; in GEMVForLstmCell() local
462 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCell()
463 bias_ptr += 4; in GEMVForLstmCell()
505 const int32* bias_ptr = bias_data; in GEMVForLstmCellWithSymmetricRange() local
515 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCellWithSymmetricRange()
516 bias_ptr += 4; in GEMVForLstmCellWithSymmetricRange()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc654 auto bias_ptr = AsDeviceMemory(bias.template flat<BiasType>().data(), in launch() local
725 side_input_scale, bias_desc, bias_ptr, dnn_activation_mode, in launch()
759 bias_desc, bias_ptr, dnn_activation_mode, output_desc, in launch()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h742 se::DeviceMemory<T> bias_ptr = tensor_on_device(bias);
786 bias_desc, bias_ptr, // bias
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h440 const int32* bias_ptr = bias_data; in GEMVForLstmCell() local
584 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCell()
585 bias_ptr += 4; in GEMVForLstmCell()
627 const int32* bias_ptr = bias_data; in GEMVForLstmCellWithSymmetricRange() local
637 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCellWithSymmetricRange()
638 bias_ptr += 4; in GEMVForLstmCellWithSymmetricRange()
988 const int32* bias_ptr = bias_data; in FullyConnectedAsGEMV() local
1089 int32x4_t bias_vec = vld1q_s32(bias_ptr); in FullyConnectedAsGEMV()
1090 bias_ptr += 4; in FullyConnectedAsGEMV()