Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru.cc31 const TfLiteTensor* gate_weight, const TfLiteTensor* gate_bias, in GruImpl() argument
47 const RuntimeShape gate_weight_shape = GetTensorShape(gate_weight); in GruImpl()
48 const float* gate_weight_data = GetTensorData<float>(gate_weight); in GruImpl()
146 const TfLiteTensor* gate_weight; in Prepare() local
148 GetInputSafe(context, node, kGateWeight, &gate_weight)); in Prepare()
149 TF_LITE_ENSURE_EQ(context, gate_weight->dims->size, 2); in Prepare()
150 TF_LITE_ENSURE_EQ(context, gate_weight->dims->data[0], 2 * n_output); in Prepare()
151 TF_LITE_ENSURE_EQ(context, gate_weight->dims->data[1], n_input + n_output); in Prepare()
231 const TfLiteTensor* gate_weight; in Eval() local
233 GetInputSafe(context, node, kGateWeight, &gate_weight)); in Eval()
[all …]
Dgru_cell.cc34 const RuntimeShape& gate_weight_shape, const float* gate_weight, in GruCell() argument
64 gate_weight, gate_bias_shape, gate_bias, activation_shape, in GruCell()
Dgru_cell.h29 const RuntimeShape& gate_weight_shape, const float* gate_weight,