Searched refs:sparse_layer_norm_lstm (Results 1 – 1 of 1) sorted by relevance
2390 HybridSparseLSTMOpModel* sparse_layer_norm_lstm, in VerifyGoldens() argument2394 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() local2699 sparse_layer_norm_lstm.SetInputGateBias(input_gate_bias_); in TEST_F()2700 sparse_layer_norm_lstm.SetCellBias(cell_gate_bias_); in TEST_F()[all …]