Searched defs:cell_to_output_weights (Results 1 – 10 of 10) sorted by relevance
/external/ComputeLibrary/arm_compute/runtime/common/ |
D | LSTMParams.h | 114 LSTMParams &set_peephole_params(T *cell_to_forget_weights, T *cell_to_output_weights) in set_peephole_params() 222 T *cell_to_output_weights() const in cell_to_output_weights() function
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | unidirectional_sequence_lstm.cc | 134 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local 513 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local 1264 const TfLiteTensor* cell_to_output_weights = GetOptionalInputTensor( in Eval() local
|
D | lstm.cc | 215 const TfLiteTensor* cell_to_output_weights = in PopulateQuantizedLstmParams8x8_16() local 503 const TfLiteTensor* cell_to_output_weights = in PopulateQuantizedLstmParams8x8_8() local 960 const TfLiteTensor* cell_to_output_weights = in CheckInputTensorDimensions() local 1866 const TfLiteTensor* cell_to_output_weights = in Eval() local
|
D | lstm_eval.cc | 1698 const TfLiteTensor* cell_to_output_weights, in EvalFloat() 1887 const TfLiteTensor* cell_to_output_weights, in EvalHybrid() 2161 const TfLiteTensor* cell_to_output_weights, in EvalInteger8x8_16() 2399 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 | 3238 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()
|