Home
last modified time | relevance | path

Searched refs:projection_bias_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Doptional_tensor_test.cc87 projection_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
89 projection_bias_ = AddNullInput(); in LSTMOpModel()
93 projection_bias_ = AddNullInput(); in LSTMOpModel()
176 PopulateTensor(projection_bias_, f); in SetProjectionBias()
216 int projection_bias_; member in tflite::__anon4dd1434f0111::LSTMOpModel
Dlstm_test.cc98 projection_bias_ = AddInput({TensorType_FLOAT32, {n_output}}); in LSTMOpModel()
100 projection_bias_ = AddNullInput(); in LSTMOpModel()
228 PopulateTensor(projection_bias_, f); in SetProjectionBias()
269 int projection_bias_; member in tflite::__anonfa79752f0111::LSTMOpModel
1408 projection_bias_ = AddInput( in LSTMIntegerOpModel()
1411 projection_bias_ = AddNullInput(); in LSTMIntegerOpModel()
1549 QuantizeAndPopulate<int32_t>(projection_bias_, f); in SetProjectionBias()
1588 int projection_bias_; member in tflite::__anonfa79752f0111::LSTMIntegerOpModel
2207 projection_bias_ = AddInput(::tflite::TensorType_FLOAT32); in HybridSparseLSTMOpModel()
2209 projection_bias_ = AddNullInput(); in HybridSparseLSTMOpModel()
[all …]
Dunidirectional_sequence_lstm_test.cc94 projection_bias_ = AddInput(TensorType_FLOAT32); in UnidirectionalLSTMOpModel()
96 projection_bias_ = AddNullInput(); in UnidirectionalLSTMOpModel()
100 projection_bias_ = AddNullInput(); in UnidirectionalLSTMOpModel()
201 PopulateTensor(projection_bias_, f); in SetProjectionBias()
255 int projection_bias_; member in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
380 std::vector<float> projection_bias_; member in tflite::__anond8582fe00111::BaseUnidirectionalLstmTest
2350 projection_bias_ = {0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, in SetUp()
2462 lstm.SetProjectionBias(projection_bias_); in TEST_F()
2500 std::vector<float> projection_bias_; member in tflite::__anond8582fe00111::BaseLayerNormUnidirectionalLstmTest
2840 projection_bias_ = AddInput( in UnidirectionalSequenceLSTMIntegerOpModel()
[all …]
Dlstm_eval_test.cc303 std::vector<int32_t> projection_bias_ = { member in tflite::__anon39d466b60111::BaseLstmParam
377 PackWeightToTensor(&projection_bias_tensor_, projection_bias_, in GetProjectionBias()
379 projection_bias_tensor_.data.i32 = projection_bias_.data(); in GetProjectionBias()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3150 projection_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
3152 projection_bias_ = AddNullInput(); in LSTMOpModel()
3156 projection_bias_ = AddNullInput(); in LSTMOpModel()
3259 PopulateTensor(projection_bias_, f); in SetProjectionBias()
3312 int projection_bias_; member in tflite::__anonde5f765e0111::LSTMOpModel