Home
last modified time | relevance | path

Searched refs:n_cell_ (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval_test.cc148 const int n_cell_ = 10; member in tflite::__anonb312a86f0111::BaseLstmParam
167 std::vector<int32_t> i2i_size_ = {n_cell_, n_input_};
183 std::vector<int32_t> i2f_size_ = {n_cell_, n_input_};
199 std::vector<int32_t> i2c_size_ = {n_cell_, n_input_};
215 std::vector<int32_t> i2o_size_ = {n_cell_, n_input_};
227 std::vector<int32_t> r2i_size_ = {n_cell_, n_output_};
239 std::vector<int32_t> r2f_size_ = {n_cell_, n_output_};
251 std::vector<int32_t> r2c_size_ = {n_cell_, n_output_};
263 std::vector<int32_t> r2o_size_ = {n_cell_, n_output_};
266 std::vector<int32_t> layer_norm_input_size_ = {n_cell_};
[all …]
Dlstm_test.cc2147 n_cell_(n_cell), in HybridSparseLSTMOpModel()
2220 ::tflite::TensorType_FLOAT32, {n_cell_ * n_batch_}}); in HybridSparseLSTMOpModel()
2300 int num_cells() { return n_cell_; } in num_cells()
2339 int n_cell_; member in tflite::__anonb3707f480111::HybridSparseLSTMOpModel
2378 int n_cell_; member in tflite::__anonb3707f480111::BaseSparseLstmTest
2428 n_cell_ = 4; in SetUp()
2659 n_batch_, n_input_, n_cell_, n_output_, in TEST_F()
2676 {n_cell_}, // cell_to_input_weight tensor in TEST_F()
2677 {n_cell_}, // cell_to_forget_weight tensor in TEST_F()
2678 {n_cell_}, // cell_to_output_weight tensor in TEST_F()
[all …]
Doptional_tensor_test.cc37 n_cell_(n_cell), in LSTMOpModel()
100 AddVariableInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}); in LSTMOpModel()
191 int num_cells() { return n_cell_; } in num_cells()
224 int n_cell_; member in tflite::__anon67d22bc80111::LSTMOpModel
Dunidirectional_sequence_lstm_test.cc43 n_cell_(n_cell), in UnidirectionalLSTMOpModel()
107 AddVariableInput(TensorData{TensorType_FLOAT32, {n_cell_ * n_batch_}}); in UnidirectionalLSTMOpModel()
229 int num_cells() { return n_cell_; } in num_cells()
269 int n_cell_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc193 SmallVector<int64_t, 2> begin_i2i_values = {n_cell_, 0}; in SetWeightForInputToInputGate()
204 int input_forget_start = couple_input_forget_gates_ ? n_cell_ : 2 * n_cell_; in SetWeightForInputToForgetGate()
214 couple_input_forget_gates_ ? 2 * n_cell_ : 3 * n_cell_; in SetWeightForInputToOutputGate()
231 SmallVector<int64_t, 2> begin_rec2i_values = {n_cell_, n_input_}; in SetWeightForRecurrentToInputGate()
242 int rec_forget_start = couple_input_forget_gates_ ? n_cell_ : 2 * n_cell_; in SetWeightForRecurrentToForgetGate()
251 int rec_output_start = couple_input_forget_gates_ ? 2 * n_cell_ : 3 * n_cell_; in SetWeightForRecurrentToOutputGate()
267 SmallVector<int64_t, 1> begin_bias2i_values = {n_cell_}; in SetBiasToInputGate()
277 int bias_forget_start = couple_input_forget_gates_ ? n_cell_ : 2 * n_cell_; in SetBiasToForgetGate()
286 couple_input_forget_gates_ ? 2 * n_cell_ : 3 * n_cell_; in SetBiasToOutputGate()
296 SmallVector<int64_t, 2> projection_slice_size_values = {n_output_, n_cell_}; in SetProjection()
[all …]
Dlstm_utils.h117 int n_cell_; variable
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3099 n_cell_(n_cell), in LSTMOpModel()
3287 int num_cells() { return n_cell_; } in num_cells()
3327 int n_cell_; member in tflite::__anon0c68fcb70111::LSTMOpModel