signature_test.cc | 255 output_tensor_names = output_tensor_names_arg; 271 std::vector<string> output_tensor_names; member in struct:tensorflow::serving::__anon44780::MockSession 318 ASSERT_EQ(2, session_.output_tensor_names.size()); 319 EXPECT_EQ(kClassesName, session_.output_tensor_names[0]); 320 EXPECT_EQ(kScoresName, session_.output_tensor_names[1]); 339 ASSERT_EQ(1, session_.output_tensor_names.size()); 340 EXPECT_EQ(kClassesName, session_.output_tensor_names[0]); 359 ASSERT_EQ(1, session_.output_tensor_names.size()); 360 EXPECT_EQ(kScoresName, session_.output_tensor_names[0]); 445 ASSERT_EQ(1, session_.output_tensor_names.size()) [all...] |