- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 34 for NextIteration (0.26 sec)
-
tensorflow/cc/ops/while_loop.cc
result[i] = outputs[i].node(); } return result; } // Manually generates the name of the `loop_var_idx`-th NextIteration node of a // loop being constructed with `scope`. This is used to define the backedge // before the NextIteration node is created. string NextIterationName(const Scope& scope, int loop_var_idx) { string result; const string& prefix = scope.impl()->name();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 01:01:21 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_executor_ops.mlir
} // CHECK-LABEL: func @nextiteration(%{{.*}}: tensor<*xf32>, %{{.*}}: i1) -> tensor<*xf32> { func.func @nextiteration(%arg0: tensor<*xf32>, %arg1: i1) -> tensor<*xf32> { %0 = tf_executor.graph { %1:3 = tf_executor.NextIteration.Source : tensor<*xf32> tf_executor.NextIteration.Sink[%1#1] %1#0 : tensor<*xf32> // CHECK: tf_executor.NextIteration.Source : tensor<*xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 25.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
return (!lhs_device_attr && !rhs_device_attr) || (lhs_device_attr && rhs_device_attr && lhs_device_attr.getValue() == rhs_device_attr.getValue()); }; // Check if tf_executor.NextIteration.Source/tf_executor.NextIteration.Sink // pair has matching devices or no devices. if (auto source = llvm::dyn_cast<tf_executor::NextIterationSourceOp>(op)) { return ops_have_same_device(source, source.GetSink());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/jit/resource_operation_safety_analysis.cc
// | | // v | // NextIteration --+ // // we won't put (Read, Write) in the returned set. This is fine if // auto-clustering can only cluster the Read->Write edge, but it is a problem if // it clusters the Write->NextIteration->Merge->Read edges instead. So we rely // on auto-clustering to not cluster NextIteration->Merge edges. The same
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference_test.cc
scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); auto add = ops::Add(scope.WithOpName("while/add"), identity_reshaped, one); auto next_iteration = ops::NextIteration(scope.WithOpName("while/NextIteration"), add); auto sink = ops::Identity(scope.WithOpName("sink"), exit); // Remove the dummy node and add the loop backedge. scope.graph()->RemoveNode(dummy.node());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
} } return absl::OkStatus(); } // Returns true if the executor/control dialect op should map to Ref node in // TensorFlow Graph. For control dialect NextIteration it uses the 1st operand // type. For executor dialect NextIteration it uses the 2nd operand type. For // all others (Enter/Exit/Merge/Switch), if the output type is ref, they // correspond to the Ref equivalent op in TF Graph.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/graph_pruning.cc
}); } VisitOpOperands(graph, op, reachable_ops, ops_to_visit); // If op is a `tf_executor.NextIteration.Source`, visit its associated // `tf_executor.NextIteration.Sink` op. if (auto source_op = llvm::dyn_cast<NextIterationSourceOp>(op)) { Operation* sink_op = source_op.GetSink().getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/executor_island_coarsening.mlir
%1:2 = tf_executor.Enter %0#0 frame "while/while_context" : (tensor<i32>) -> (tensor<*xi32>, !tf_executor.control) %2 = tf_executor.island wraps "tf.opB"() : () -> () %3:3 = tf_executor.NextIteration.Source : tensor<*xi32> %4:3 = tf_executor.Merge %3#0, %1#0 : tensor<*xi32> %5:2 = tf_executor.island(%4#2) wraps "tf.opC"() : () -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 17.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
if (auto input_result = mlir::dyn_cast<mlir::OpResult>(src)) { auto* input_inst = GetIslandInnerOpOrSelf(input_result.getOwner()); // Replaces the input node with NextIteration sink if it is a NextIteration // source. if (auto next_iter_source = llvm::dyn_cast<mlir::tf_executor::NextIterationSourceOp>( input_inst)) input_inst = next_iter_source.GetSink();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
if (auto input_result = mlir::dyn_cast<mlir::OpResult>(src)) { auto* input_inst = GetIslandInnerOpOrSelf(input_result.getOwner()); // Replaces the input node with NextIteration sink if it is a NextIteration // source. if (auto next_iter_source = llvm::dyn_cast<mlir::tf_executor::NextIterationSourceOp>( input_inst)) input_inst = next_iter_source.GetSink();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0)