Home
last modified time | relevance | path

Searched refs:candidate_weight (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc32 const TfLiteTensor* candidate_weight, in GruImpl() argument
51 const RuntimeShape candidate_weight_shape = GetTensorShape(candidate_weight); in GruImpl()
52 const float* candidate_weight_data = GetTensorData<float>(candidate_weight); in GruImpl()
161 const TfLiteTensor* candidate_weight; in Prepare() local
163 &candidate_weight)); in Prepare()
164 TF_LITE_ENSURE_EQ(context, candidate_weight->dims->size, 2); in Prepare()
165 TF_LITE_ENSURE_EQ(context, candidate_weight->dims->data[0], n_output); in Prepare()
166 TF_LITE_ENSURE_EQ(context, candidate_weight->dims->data[1], in Prepare()
237 const TfLiteTensor* candidate_weight; in Eval() local
239 &candidate_weight)); in Eval()
[all …]
Dgru_cell.cc37 const float* candidate_weight, in GruCell() argument
81 candidate_weight, candidate_bias_shape, candidate_bias, in GruCell()
Dgru_cell.h32 const float* candidate_weight,