Searched defs:cell_to_forget_weights (Results 1 – 10 of 10) sorted by relevance
| /external/ComputeLibrary/arm_compute/runtime/common/ |
| D | LSTMParams.h | 112 LSTMParams &set_peephole_params(T *cell_to_forget_weights, T *cell_to_output_weights) in set_peephole_params()
215 T *cell_to_forget_weights() const in cell_to_forget_weights() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 133 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
504 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local
1266 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in Eval() local
|
| D | lstm.cc | 214 const TfLiteTensor* cell_to_forget_weights = in PopulateQuantizedLstmParams8x8_16() local
502 const TfLiteTensor* cell_to_forget_weights = in PopulateQuantizedLstmParams8x8_8() local
951 const TfLiteTensor* cell_to_forget_weights = in CheckInputTensorDimensions() local
1868 const TfLiteTensor* cell_to_forget_weights = in Eval() local
|
| D | lstm_eval.cc | 1755 const TfLiteTensor* cell_to_forget_weights, in EvalFloat()
1950 const TfLiteTensor* cell_to_forget_weights, in EvalHybrid()
2224 const TfLiteTensor* cell_to_forget_weights, in EvalInteger8x8_16()
2462 const TfLiteTensor* cell_to_forget_weights, in EvalInteger8x8_8()
|
| D | bidirectional_sequence_lstm.cc | 290 const TfLiteTensor* cell_to_forget_weights = in CheckLstmTensorDimensionsAndTypes() local
|
| D | unidirectional_sequence_lstm_test.cc | 2984 const std::vector<float> cell_to_forget_weights = {0.2, -0.1, 0.1, -0.2}; in TEST() local
|
| D | lstm_test.cc | 1804 const std::vector<float> cell_to_forget_weights = {0.2, -0.1, 0.1, -0.2}; in TEST() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 350 const TfLiteTensor* cell_to_forget_weights, in EvalFloat()
571 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 274 const TfLiteTensor* cell_to_forget_weights, in EvalCalibration()
516 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 49 std::initializer_list<float> cell_to_forget_weights, in LSTMOpModel()
|