Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for NextIteration (0.16 sec)

  1. tensorflow/compiler/jit/deadness_analysis.cc

        }
      }
      return errors::InvalidArgument(
          "Multiple NextIteration inputs to merge node ",
          FormatNodeForError(*merge), ": \n", absl::StrJoin(backedges, "\n"),
          "\nMerge nodes can have at most one incoming NextIteration edge.");
    }
    
    Status FindUniqueBackedge(Node* merge, const Edge** result) {
      *result = nullptr;
      CHECK(merge->IsMerge());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 60.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/deadness_analysis_test.cc

                               latch.output_false);
      Output iv_next = ops::Add(root.WithOpName(prefix + "/ivnext"),
                                latch.output_true, increment_by);
      Output next_iteration =
          ops::NextIteration(root.WithOpName(prefix + "/next_iteration"), iv_next);
    
      CHECK(root.graph()
                ->UpdateEdge(next_iteration.node(), 0, iv.output.node(), 1)
                .ok());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 51.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

      auto b = ops::Placeholder(root.WithOpName("B"), DT_FLOAT);
      auto c = ops::Add(root.WithOpName("C"), a, b);
      auto enter = ops::internal::Enter(root, c, "aframe");
      auto next_iter = ops::NextIteration(root, enter);
      auto exit = ops::internal::Exit(root, next_iter);
      auto d = ops::Add(root.WithOpName("D"), c, exit);
    
      std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        // and only disallow these Identities to go into the cluster with nodes from
        // the loop body but still consider them candidates.
        //
        // LoopCond ->
        // Merge    -> Switch -> Identity -> i++ -> ... -> NextIteration
        //                               ..> Const -> LoopBody
        //                            (control edge)
        TF_ASSIGN_OR_RETURN(bool is_identity_driving_consts_in_loop,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top