Searched defs:cell_to_output_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() 220 T *cell_to_output_weights() const in cell_to_output_weights() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 135 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local 514 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local 1268 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in Eval() local
|
| D | lstm.cc | 216 const TfLiteTensor* cell_to_output_weights = in PopulateQuantizedLstmParams8x8_16() local 504 const TfLiteTensor* cell_to_output_weights = in PopulateQuantizedLstmParams8x8_8() local 961 const TfLiteTensor* cell_to_output_weights = in CheckInputTensorDimensions() local 1870 const TfLiteTensor* cell_to_output_weights = in Eval() local
|
| D | lstm_eval.cc | 1756 const TfLiteTensor* cell_to_output_weights, in EvalFloat() 1951 const TfLiteTensor* cell_to_output_weights, in EvalHybrid() 2225 const TfLiteTensor* cell_to_output_weights, in EvalInteger8x8_16() 2463 const TfLiteTensor* cell_to_output_weights, in EvalInteger8x8_8()
|
| D | bidirectional_sequence_lstm.cc | 299 const TfLiteTensor* cell_to_output_weights = in CheckLstmTensorDimensionsAndTypes() local
|
| D | unidirectional_sequence_lstm_test.cc | 2986 const std::vector<float> cell_to_output_weights = {0.3, -0.1, 0.1, -0.3}; in TEST() local
|
| D | lstm_test.cc | 1806 const std::vector<float> cell_to_output_weights = {0.3, -0.1, 0.1, -0.3}; in TEST() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 351 const TfLiteTensor* cell_to_output_weights, in EvalFloat() 573 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in lstm_eval() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 275 const TfLiteTensor* cell_to_output_weights, in EvalCalibration() 518 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in lstm_eval() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 50 std::initializer_list<float> cell_to_output_weights, in LSTMOpModel()
|