Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for IfRegion (0.57 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/merge_control_flow.mlir

    // can be merged but not 2nd IfRegion and 2nd IfRegion should be moved after
    // newly merged IfRegion.
    
    // CHECK-LABEL: func @same_predicate_3_ifregions_reorder
    func.func @same_predicate_3_ifregions_reorder() {
      // CHECK:      tf_device.cluster
      // CHECK:        "tf.IfRegion"
      // CHECK:          "tf.A"
      // CHECK:          "tf.G"
      // CHECK-NEXT    "tf.IfRegion"
      // CHECK:          "tf.E"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 63.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/resource_op_lifting.mlir

      // CHECK: %[[IF:.*]]:2 = "tf.IfRegion"(%[[ARG0]])
      %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf_type.resource<tensor<4xf32>>>
      %1 = "tf.VarHandleOp"() {container = "c", shared_name = "v2"} : () -> tensor<*x!tf_type.resource<tensor<4xf32>>>
      %2 = "tf_device.cluster"() ({
        %3:2 = "tf.IfRegion"(%arg0) ({
              // CHECK-NEXT: %[[CONST:.*]] = "tf.Const"()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 74K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/tensor_array_ops_decomposition.mlir

      // CHECK: "tf.IfRegion"(%[[PRED]]) <{is_stateless = false}> ({
      %case_op = "tf.IfRegion"(%arg0) <{is_stateless = false}> ({
          // CHECK: %[[TA_VAL:.*]] = "tf.ReadVariableOp"(%[[TA_BUFFER]])
          // CHECK: "tf.Slice"(%[[TA_VAL]]
          // CHECK-NOT: tf.TensorArrayReadV3
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 49K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md

    their `mlir_module` string attribute.
    ### `-tf-merge-control-flow`
    
    _Merges IfRegion ops together with a common predicate._
    
    This pass merges IfRegion ops together if they have the same predicate and it
    is safe to do so (there are no intermediate dependencies, they are in the
    same block, etc).
    
    For example:
    
    ```mlir
    "tf.IfRegion"(%0) ( {
      %2 = "tf.A"() : () -> (tensor<f32>)
      "tf.Yield"() : () -> ()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 02 02:26:39 UTC 2023
    - 96.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

      std::string retvals_communication_key =
          llvm::formatv("host_compute_channel_{0}_retvals",
                        (communication_key_index))
              .str();
    
      // Use a unique name when sending just the IfRegion predicate.  This is
      // for readable and to match the key in the TF2XLA bridge.
      if (clustered_ops.size() == 1 && llvm::isa<mlir::TF::IfRegionOp>(op) &&
          external_operands.size() == 1) {
        args_communication_key =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

        types of inputs must match the signature of the operation that contains the
        region.
      }];
    
      let arguments = (ins Variadic<AnyType>);
    }
    
    def TF_IfRegionOp : TF_Op<"IfRegion",
          [SingleBlockImplicitTerminator<"YieldOp">, NoRegionArguments,
           DeclareOpInterfaceMethods<RegionBranchOpInterface, [
               "areTypesCompatible",
               "getEntrySuccessorOperands",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
Back to top