HomeSort by relevance Sort by last modified time
    Searched defs:output_tensor_names (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/cc/client/
client_session.cc 115 std::vector<string> output_tensor_names; local
116 output_tensor_names.reserve(fetch_outputs.size());
118 output_tensor_names.push_back(output.name());
126 return impl()->session_->Run(run_options, feeds, output_tensor_names,
  /external/tensorflow/tensorflow/contrib/session_bundle/
signature.cc 118 std::vector<string> output_tensor_names; local
120 output_tensor_names.push_back(signature.classes().tensor_name());
123 output_tensor_names.push_back(signature.scores().tensor_name());
129 output_tensor_names, {}, &outputs);
136 if (outputs.size() != output_tensor_names.size()) {
138 strings::StrCat("Expected ", output_tensor_names.size(),
173 std::vector<string> output_tensor_names; local
175 output_tensor_names.push_back(signature.output().tensor_name());
181 output_tensor_names, {}, &outputs);
188 if (outputs.size() != output_tensor_names.size())
    [all...]
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...]
  /external/tensorflow/tensorflow/tools/benchmark/
benchmark_model.cc 122 std::vector<string> output_tensor_names; local
134 output_tensor_names.push_back(wanted_shape);
138 session->Run(input_tensors, output_tensor_names, {}, &output_tensors));
139 CHECK_EQ(output_tensors.size(), output_tensor_names.size());
140 for (int i = 0; i < output_tensor_names.size(); ++i) {
141 const string& wanted_shape_name = output_tensor_names[i];

Completed in 808 milliseconds