Function TEST

tensorflow/core/graph/control_flow_test.cc:48–75

}

TEST(ValidateControlFlowTest, InputsFromDifferentFrames) {
  Scope scope = Scope::NewRootScope().ExitOnError();
  std::vector<Output> inputs;
  inputs.push_back(ops::Placeholder(scope, DT_INT32));
  std::vector<Output> outputs;
  TF_ASSERT_OK(ops::BuildWhileLoop(scope.NewSubScope("outer"), inputs,
                                   LessThanTenCond, NestedLoopBody,
                                   "outer_loop", &outputs));
  std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()));
  TF_ASSERT_OK(scope.ToGraph(graph.get()));
  // {inner/Enter', 'outer/Switch'} --> 'inner/Merge'. 'inner/Enter' is in frame
  // 'inner_loop'. 'outer/Switch' is in frame 'outer_loop'.
  std::vector<ControlFlowInfo> info;
  Status status = BuildControlFlowInfo(graph.get(), &info);
  EXPECT_FALSE(status.ok());
  EXPECT_TRUE(absl::StrContains(status.error_message(),
                                "has inputs from different frames"))
      << status.error_message();
  EXPECT_TRUE(absl::StrContains(status.error_message(),
                                "{{node outer/body/inner/Merge}}"))
      << status.error_message();
  EXPECT_TRUE(absl::StrContains(status.error_message(),
                                "{{node outer/body/inner/Enter}}"))
      << status.error_message();
  EXPECT_TRUE(
      absl::StrContains(status.error_message(), "{{node outer/Switch}}"))
      << status.error_message();
}

TEST(ValidateControlFlowTest, MismatchedParentFrames) {
  Scope scope = Scope::NewRootScope().ExitOnError();