Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 37 for NextIteration (0.8 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top