Function TEST_F

tensorflow/core/graph/optimizer_fusion_engine_test.cc:129–201

REGISTER_OP("Output").Output("o: float");

TEST_F(OptimizerFusionTest, test_input_is_control_dependency_edge) {
  GraphDef test_graph;
  AttrValue type;
  type.set_type(DT_FLOAT);
  AddNode("begin1", "Const", {}, {{"dtype", type}}, &test_graph);
  AddNode("s1", "Identity", 
      {AsControlDependency("begin1")}, 
      {{"T", type}}, &test_graph);
  AddNode("s2", "Identity", {"s1"}, {{"T", type}}, &test_graph);

  GraphConstructorOptions opts;
  TF_CHECK_OK(ConvertGraphDefToGraph(opts, test_graph, &graph_));

  class TemplateControlDeps : public TemplateBase {
   public:
    TemplateControlDeps() {
      const TempNode n0 = {
        .key = "begin1",
        .op = "Const",
        .inputs = {},
        .outputs = {{}},
        .deps_inputs = {},
        .deps_outputs = {"s1"}
      };
      temp_nodes_.emplace_back(n0);
      const TempNode n2 = {
        .key = "s1",
        .op = "Identity",
        .inputs = {},
        .outputs = {{"0"}},
        .deps_inputs = {"begin1", "end1"}
      };
      temp_nodes_.emplace_back(n2);

      first_key_ = "begin1";
      num_inputs_ = 0;
      num_outputs_ = 1;
    }

    bool add_subgraph(std::map<std::string, MatchedNode>& nodes,
        std::string name_prefix, Graph* g,
        std::vector<const Edge*>& inputs,
        std::vector<std::vector<const Edge*>>& outputs) override {
      NodeDef fused_def;
      fused_def.set_op("Output");
      fused_def.set_name("fused_op");
      Status status;
      Node* fused_node = g->AddNode(fused_def, &status);
      add_oedges(g, fused_node, 0, outputs[0]);
      return true;
    }

    bool CheckDynamicInputs(
        const Node* node, const TempNode* temp_node, int dy_mode,
        std::vector<const Edge*>& fused_op_inputs,
        std::map<const std::string, TempNode>& temp_node_map,
        std::map<std::string, MatchedNode>& matched_node_map) override {
      return false;