experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TFL: Make the test more robust.

Browse Source 
ANSI C requires len(argv) + 1 == argc and argv[argc] = 0

PiperOrigin-RevId: 256306649
This commit is contained in:
Tiezhen WANG 2019-07-02 23:18:34 -07:00 committed by TensorFlower Gardener
parent e0ab930a7e
commit 8719744db9
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/toco/model_cmdline_flags_test.cc
View File

@ -28,13 +28,12 @@ namespace {
TEST(ModelCmdlineFlagsTest, ParseArgsStringMapList) { TEST(ModelCmdlineFlagsTest, ParseArgsStringMapList) {
int args_count = 3; int args_count = 3;
const char* args[] = { const char* args[] = {
"toco", "toco", "--input_arrays=input_1",
"--input_arrays=input_1",
"--rnn_states={state_array:rnn/BasicLSTMCellZeroState/zeros," "--rnn_states={state_array:rnn/BasicLSTMCellZeroState/zeros,"
"back_edge_source_array:rnn/basic_lstm_cell/Add_1,size:4}," "back_edge_source_array:rnn/basic_lstm_cell/Add_1,size:4},"
"{state_array:rnn/BasicLSTMCellZeroState/zeros_1," "{state_array:rnn/BasicLSTMCellZeroState/zeros_1,"
"back_edge_source_array:rnn/basic_lstm_cell/Mul_2,size:4}", "back_edge_source_array:rnn/basic_lstm_cell/Mul_2,size:4}",
}; nullptr};
string expected_input_arrays = "input_1"; string expected_input_arrays = "input_1";
std::vector<std::unordered_map<string, string>> expected_rnn_states; std::vector<std::unordered_map<string, string>> expected_rnn_states;