Searched refs:hidden_state (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 58 const TfLiteTensor* hidden_state = in Prepare() local
74 TF_LITE_ENSURE_EQ(context, NumDimensions(hidden_state), 2); in Prepare()
75 TF_LITE_ENSURE_EQ(context, hidden_state->dims->data[0], batch_size); in Prepare()
76 TF_LITE_ENSURE_EQ(context, hidden_state->dims->data[1], num_units); in Prepare()
110 hidden_state->dims)) { in Prepare()
112 TfLiteIntArrayCopy(hidden_state->dims); in Prepare()
137 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalFloat() argument
145 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalFloat()
168 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalHybrid() argument
176 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalHybrid()
[all …]
|
| D | unidirectional_sequence_rnn.cc | 59 const TfLiteTensor* hidden_state = in Prepare() local
80 TF_LITE_ENSURE_EQ(context, NumDimensions(hidden_state), 2); in Prepare()
81 TF_LITE_ENSURE_EQ(context, hidden_state->dims->data[0], batch_size); in Prepare()
82 TF_LITE_ENSURE_EQ(context, hidden_state->dims->data[1], num_units); in Prepare()
117 hidden_state->dims)) { in Prepare()
119 TfLiteIntArrayCopy(hidden_state->dims); in Prepare()
144 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalFloat() argument
162 float* hidden_state_ptr_batch = GetTensorData<float>(hidden_state); in EvalFloat()
181 GetTensorData<float>(hidden_state) + b * num_units; in EvalFloat()
205 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalHybrid() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_ophint_func_op.cc | 103 Value hidden_state = call_op.getOperand(4); in BuildUnidirectionalSequenceRnnOp() local
116 hidden_state, builder->getBoolAttr(time_major), in BuildUnidirectionalSequenceRnnOp()
|
| D | load_quantization_recipe.cc | 185 auto hidden_state = builder->create<MulOp>( in LoadForLSTMOp() local
188 loc, int8, hidden_state.output(), lstm.projection_weights(), in LoadForLSTMOp()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | gru_v2_test.py | 609 hidden_state = array_ops.zeros([1, gru_unit_size], dtype=dtypes.float32)
610 _, state = gru(x, initial_state=hidden_state)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 3247 TFL_StatefulTensor:$hidden_state,
|