Searched refs:ok_params (Results 1 – 1 of 1) sorted by relevance
1552 TestParams ok_params[kReshapeOKCases] = { in TEST_F() local1559 AddTestTensor("input", ok_params[i].tensor_dims, ok_params[i].batch_size); in TEST_F()1560 AddTestWeights<int32>("weights", {4}, ok_params[i].shape); in TEST_F()2462 TestParams ok_params[kExpandDimsOKCases] = { in TEST_F() local2470 AddTestTensor("input", ok_params[i].input_dims); in TEST_F()2471 AddTestWeights<int32>("weights", {1}, {ok_params[i].axis}); in TEST_F()2476 ExpectTrtDimsEqualsArray(ok_params[i].expected_output_dims, in TEST_F()2574 TestParams ok_params[kSqueezeOKCases] = { in TEST_F() local2588 NodeDef node_def = get_squeeze_nodedef(ok_params[i].axis); in TEST_F()2589 AddTestTensor("input", ok_params[i].input_dims); in TEST_F()[all …]