Lines Matching refs:session_
297 MockSession session_;
302 session_.outputs = {test::AsTensor<int>({3}), test::AsTensor<int>({2})};
303 const Status status = RunClassification(signature_, input_tensor_, &session_,
312 ASSERT_EQ(1, session_.inputs.size());
313 EXPECT_EQ(kInputName, session_.inputs[0].first);
315 session_.inputs[0].second);
317 ASSERT_EQ(2, session_.output_tensor_names.size());
318 EXPECT_EQ(kClassesName, session_.output_tensor_names[0]);
319 EXPECT_EQ(kScoresName, session_.output_tensor_names[1]);
324 session_.outputs = {test::AsTensor<int>({3})};
325 const Status status = RunClassification(signature_, input_tensor_, &session_,
333 ASSERT_EQ(1, session_.inputs.size());
334 EXPECT_EQ(kInputName, session_.inputs[0].first);
336 session_.inputs[0].second);
338 ASSERT_EQ(1, session_.output_tensor_names.size());
339 EXPECT_EQ(kClassesName, session_.output_tensor_names[0]);
344 session_.outputs = {test::AsTensor<int>({2})};
345 const Status status = RunClassification(signature_, input_tensor_, &session_,
353 ASSERT_EQ(1, session_.inputs.size());
354 EXPECT_EQ(kInputName, session_.inputs[0].first);
356 session_.inputs[0].second);
358 ASSERT_EQ(1, session_.output_tensor_names.size());
359 EXPECT_EQ(kScoresName, session_.output_tensor_names[0]);
423 MockSession session_;
428 session_.outputs = {test::AsTensor<float>({1, 2})};
430 RunRegression(signature_, input_tensor_, &session_, &output_tensor_);
437 ASSERT_EQ(1, session_.inputs.size());
438 EXPECT_EQ(kInputName, session_.inputs[0].first);
440 session_.inputs[0].second);
442 ASSERT_EQ(1, session_.output_tensor_names.size());
443 EXPECT_EQ(kRegressionsName, session_.output_tensor_names[0]);
448 session_.status = errors::DataLoss("Data is gone");
450 RunRegression(signature_, input_tensor_, &session_, &output_tensor_);
458 session_.outputs = {test::AsTensor<float>({3})};
461 RunRegression(signature_, input_tensor_, &session_, &output_tensor_);
