Searched defs:projection_weights (Results 1 – 10 of 10) sorted by relevance
| /external/ComputeLibrary/arm_compute/runtime/common/ |
| D | LSTMParams.h | 98 LSTMParams &set_projection_params(const T *projection_weights, const T *projection_bias) in set_projection_params() 225 const T *projection_weights() const in projection_weights() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 147 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local 586 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local 761 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in PopulatePrecomputedZPTimesWeightsWithBias() local 1157 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in Prepare() local 1286 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in Eval() local
|
| D | lstm.cc | 228 const TfLiteTensor* projection_weights = in PopulateQuantizedLstmParams8x8_16() local 528 const TfLiteTensor* projection_weights = in PopulateQuantizedLstmParams8x8_8() local 1029 const TfLiteTensor* projection_weights = in CheckInputTensorDimensions() local 1195 const TfLiteTensor* projection_weights = in PopulatePrecomputedZPTimesWeightsWithBias() local 1605 const TfLiteTensor* projection_weights = in Prepare() local 1734 const TfLiteTensor* projection_weights = in Prepare() local 1894 const TfLiteTensor* projection_weights = in Eval() local
|
| D | lstm_eval.cc | 305 const float* projection_weights, in CalculateLstmOutputFloat() 477 const int8_t* projection_weights, const uint8_t* projection_weights_ledger, in CalculateLstmOutputHybrid() 662 const int8_t* projection_weights, int32_t proj_scale_a, in CalculateLstmOutputInteger8x8_16() 768 const int16_t* output_gate, const int8_t* projection_weights, in CalculateLstmOutputInteger8x8_8() 1768 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalFloat() 1932 TfLiteStatus EvalHybrid( in EvalHybrid() 2232 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalInteger8x8_16() 2470 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalInteger8x8_8()
|
| D | lstm_test.cc | 1647 const std::vector<float> projection_weights = { in TEST() local 1815 const std::vector<float> projection_weights = { in TEST() local 1982 const std::vector<float> projection_weights = { in TEST() local
|
| D | unidirectional_sequence_lstm_test.cc | 2831 const std::vector<float> projection_weights = { in TEST() local 2995 const std::vector<float> projection_weights = { in TEST() local
|
| D | bidirectional_sequence_lstm.cc | 353 const TfLiteTensor* projection_weights = in CheckLstmTensorDimensionsAndTypes() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 126 const float* projection_weights, const float* projection_bias, in CalculateLstmOutputCalibration() 287 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalCalibration() 552 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in lstm_eval() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 363 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalFloat() 598 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in lstm_eval() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 55 std::initializer_list<float> projection_weights, in LSTMOpModel()
|