Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/
Dkernel_utils.cc23 const float* recurrent_weights_ptr, const float* bias_ptr, in RnnBatchStep()
39 const float* recurrent_weights_ptr, const float* bias_ptr, in RnnBatchStep()
123 float recurrent_weights_scale, const float* bias_ptr, int input_size, in RnnBatchStep()
145 const float* bias_ptr, int input_size, int aux_input_size, int num_units, in RnnBatchStep()
/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc165 const float* bias_ptr = bias->data.f; in EvalFloat() local
199 const float* bias_ptr = bias->data.f; in EvalHybrid() local
Dunidirectional_sequence_rnn.cc153 const float* bias_ptr = bias->data.f; in EvalFloat() local
219 const float* bias_ptr = bias->data.f; in EvalHybrid() local
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc131 float* bias_ptr = static_cast<float*>(port::AlignedMalloc( in Compute() local
/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
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h440 const int32* bias_ptr = bias_data; in GEMVForLstmCell() local
627 const int32* bias_ptr = bias_data; in GEMVForLstmCellWithSymmetricRange() local
988 const int32* bias_ptr = bias_data; in FullyConnectedAsGEMV() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h318 const int32* bias_ptr = bias_data; in GEMVForLstmCell() local
505 const int32* bias_ptr = bias_data; in GEMVForLstmCellWithSymmetricRange() local