Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm_test.cc2390 HybridSparseLSTMOpModel* sparse_layer_norm_lstm, in VerifyGoldens() argument
2394 const int num_inputs = sparse_layer_norm_lstm->num_inputs(); in VerifyGoldens()
2403 sparse_layer_norm_lstm->SetInput( in VerifyGoldens()
2404 b * sparse_layer_norm_lstm->num_inputs(), batch_start, batch_end); in VerifyGoldens()
2407 ASSERT_EQ(sparse_layer_norm_lstm->Invoke(), kTfLiteOk); in VerifyGoldens()
2409 const int num_outputs = sparse_layer_norm_lstm->num_outputs(); in VerifyGoldens()
2417 sparse_layer_norm_lstm->GetOutput(), in VerifyGoldens()
2656 HybridSparseLSTMOpModel sparse_layer_norm_lstm( in TEST_F() local
2699 sparse_layer_norm_lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
2700 sparse_layer_norm_lstm.SetCellBias(cell_gate_bias_); in TEST_F()
[all …]