Searched refs:ok_params (Results 1 – 1 of 1) sorted by relevance
2383 std::vector<TestParams> ok_params = { in TEST_F() local2393 for (int i = 0; i < ok_params.size(); ++i) { in TEST_F()2394 const int batch_size = std::max(1, ok_params[i].batch_size); in TEST_F()2395 const auto& shape = ok_params[i].shape; in TEST_F()2397 AddTestTensor("input", ok_params[i].tensor_dims, batch_size); in TEST_F()3749 std::vector<TestParams> ok_params = { in TEST_F() local4170 for (int i = 0; i < ok_params.size(); i++) { in TEST_F()4173 ok_params[i].begin_mask, ok_params[i].end_mask, in TEST_F()4174 ok_params[i].ellipsis_mask, ok_params[i].new_axis_mask, in TEST_F()4175 ok_params[i].shrink_axis_mask); in TEST_F()[all …]