Searched refs:candidate_bias (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru.cc | 33 const TfLiteTensor* candidate_bias, TfLiteTensor* output, in GruImpl() argument
53 const RuntimeShape candidate_bias_shape = GetTensorShape(candidate_bias); in GruImpl()
54 const float* candidate_bias_data = GetTensorData<float>(candidate_bias); in GruImpl()
170 const TfLiteTensor* candidate_bias; in Prepare() local
172 context, GetInputSafe(context, node, kCandidateBias, &candidate_bias)); in Prepare()
173 TF_LITE_ENSURE_EQ(context, candidate_bias->dims->size, 1); in Prepare()
174 TF_LITE_ENSURE_EQ(context, candidate_bias->dims->data[0], n_output); in Prepare()
240 const TfLiteTensor* candidate_bias; in Eval() local
242 context, GetInputSafe(context, node, kCandidateBias, &candidate_bias)); in Eval()
257 candidate_bias, output, output_state, activation, concat, in Eval()
|
| D | gru_cell.cc | 39 const float* candidate_bias, const RuntimeShape& output_shape, in GruCell() argument
81 candidate_weight, candidate_bias_shape, candidate_bias, in GruCell()
|
| D | gru_cell.h | 34 const float* candidate_bias, const RuntimeShape& output_shape,
|