Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for else_branch (0.17 sec)

  1. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

        bool* has_outside_compilation) {
      // Instantiate "then_branch" and "else_branch".
      NameAttrList then_branch, else_branch;
      TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "then_branch", &then_branch));
      TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "else_branch", &else_branch));
    
      // Extract outside compilation for then_branch and else_branch.
      bool then_branch_has_outside_compilation = false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir

      // CHECK-SAME: device = "noodle"
      %3 = "tf.If"(%1, %2, %arg1) {then_branch = @add, else_branch = @sub, output_shapes = [#tf_type.shape<>], device = "noodle", _underscore_attr = "something", is_stateless = false} : (tensor<i1>, tensor<f32>, tensor<f32>) -> tensor<f32>
    
      // CHECK: %2 = "tf.If"
      %4 = "tf.If"(%arg2, %3, %arg1) {then_branch = @add, else_branch = @sub, is_stateless = false} : (tensor<i1>, tensor<f32>, tensor<f32>) -> tensor<f32>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 132.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

        `tf.IfRegion` and `tf.While` is transformed to `tf.WhileRegion`.
    
        For example, this functional operation
    
        ```mlir
          %0 = "tf.If"(%arg0, %arg1) {
            then_branch = @then_branch_func, else_branch = @else_branch_func, is_stateless = false
          } : (tensor<i1>, tensor<*xf32>) -> tensor<*xf32>
        ```
    
        will be transformed into this region-based operation
    
        ```mlir
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir

        %1 = "tf.If"(%arg0, %arg2) {Tcond = i1, Tin = ["tfdtype$DT_FLOAT"], Tout = ["tfdtype$DT_FLOAT"], _xla_propagate_compile_time_consts = true, device = "", else_branch = @reused_if_else_branch, is_stateless = true, name = "if", then_branch = @reused_if_then_branch}...
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 23 17:24:10 UTC 2024
    - 167.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

    LogicalResult IfOp::verifySymbolUses(SymbolTableCollection& symbol_table) {
      auto branch_name = [](unsigned index) -> std::string {
        return index == 0 ? "'then_branch'" : "'else_branch'";
      };
      return VerifyCaseOrIfOpBranchFunctions(
          symbol_table, *this, {getThenBranchAttr(), getElseBranchAttr()},
          branch_name);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
Back to top