Searched refs:output_gate (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 337 Value* output_state, Value* output_gate, in BuildOutputStateUpdate() argument
367 RETURN_IF_ERROR(graph->AddConsumer(node->id, output_gate->id)); in BuildOutputStateUpdate()
516 Value* output_gate; in ParseLSTMAttributes() local
524 kTfLiteActSigmoid, has_peephole, has_normalization, &output_gate)); in ParseLSTMAttributes()
528 output_gate, new_cell_state, in ParseLSTMAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 178 auto output_gate = CreateGate( in LoadForLSTMOp() local
187 loc, int16, new_cell_tanh.output(), output_gate->getResult(0), none_af); in LoadForLSTMOp()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.cc | 263 const float* cell_state, const float* output_gate, in CalculateLstmOutputFloat() argument
271 tensor_utils::VectorVectorCwiseProduct(output_gate, scratch, n_batch * n_cell, in CalculateLstmOutputFloat()
434 const float* output_gate, TfLiteFusedActivation activation, in CalculateLstmOutputHybrid() argument
443 tensor_utils::VectorVectorCwiseProduct(output_gate, scratch0, in CalculateLstmOutputHybrid()
618 int32_t cell_state_scale, const int16_t* output_gate, in CalculateLstmOutputInteger8x8_16() argument
628 tensor_utils::CwiseMul(output_gate, scratch0, hidden_scale_a, hidden_scale_b, in CalculateLstmOutputInteger8x8_16()
726 const int16_t* output_gate, const int8_t* projection_weights, in CalculateLstmOutputInteger8x8_8() argument
732 tensor_utils::CwiseMul(output_gate, scratch, n_batch, n_cell, 15 + 15 - 15, in CalculateLstmOutputInteger8x8_8()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 125 const float* output_gate, TfLiteFusedActivation activation, in CalculateLstmOutputCalibration() argument
131 tensor_utils::VectorVectorCwiseProduct(output_gate, scratch, n_batch * n_cell, in CalculateLstmOutputCalibration()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 539 const float output_gate = in LstmCell() local
549 output_gate * std::tanh(new_state); in LstmCell()
|