- Sort Score
- Result 10 results
- Languages All
Results 31 - 34 of 34 for NextIteration (0.14 sec)
-
tensorflow/compiler/jit/xla_cluster_util.cc
// graph. // * the body of the loop should now be disconnected from the rest of the // graph; we make it acyclic by breaking loop backedges (edges outgoing from // "NextIteration" nodes. // Map from frame name strings to node IDs in the cycle detection graph. std::unordered_map<string, int> frame_nodes; // Get the cycle graph node ID for frame 'frame_name', or add one if none
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 21.3K bytes - Viewed (0) -
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) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
func.func @next_iteration_user(%arg0: tensor<32x?x256x4xf32>) -> tensor<?x?x?xf32> { %0 = tf_executor.graph { // CHECK: tf_executor.NextIteration.Source // CHECK-SAME: : tensor<32x?x4xf32> %1:3 = tf_executor.NextIteration.Source : tensor<?x?x?xf32> %out, %c_out = tf_executor.island { %dims = "tf.Const"() {value = dense<[32, -1, 4]> : tensor<3xi32>} : () -> tensor<3xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
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)