- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 18 for NextIteration (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_executor_ops.td
} def TfExecutor_NextIterationSourceOp : TfExecutor_Op<"NextIteration.Source", [HasParent<"GraphOp">]> { let summary = [{ The "tf_executor.NextIteration.Source" is paired with a "tf_executor.NextIteration.sink" to represent NextIteration op in Tensorflow. }]; let description = [{ Tensorflow NextIteration operation forwards its input to the next iteration
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 23 19:35:12 UTC 2023 - 22K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_device_propagation.mlir
} func.return } // Tests tf_executor.NextIteration.Source/tf_executor.NextIteration.Sink has a // device when an intermediate op in its loop has a device. // CHECK-LABEL: func @testNextIterationNoDevice func.func @testNextIterationNoDevice() { tf_executor.graph { // CHECK: tf_executor.NextIteration.Source // CHECK-SAME: device = "/job:localhost/replica:0/task:0/device:TPU:0"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 19K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_executor_ops_invalid.mlir
%0 = tf_executor.graph { %1:3 = tf_executor.NextIteration.Source : tensor<*xf32> "tf_executor.NextIteration.Sink"(%1#1, %arg1) : (!tf_executor.token, i1) -> () // expected-error@-1 {{'tf_executor.NextIteration.Sink' op input type 'i1' mismatch the tf_executor.NextIteration.Source output type: 'tensor<*xf32>'}} tf_executor.fetch %1#0 : tensor<*xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 19 01:12:10 UTC 2023 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/g3doc/tf_dialects.md
defined as a non-dead input (selected in a non-defined way if multiple inputs are non-dead). If all inputs are dead, the output is also dead. ### NextIteration: `tf_executor.NextIteration.Source` and `tf_executor.NextIteration.Sink` Operation The TensorFlow [`NextIteration`](https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/next-iteration) op is modeled using these two paired operations. Since _NextIteration_ is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 13 16:33:28 UTC 2021 - 16K 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/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)