Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 53 of 53 for else_branch (0.16 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

            mlir::SymbolRefAttr::get(builder.getContext(), func_names.at(else_idx));
    
        return llvm::SmallVector<mlir::NamedAttribute, 4>{
            builder.getNamedAttr("then_branch", then_attr),
            builder.getNamedAttr("else_branch", else_attr),
            // TODO(b/139667752): Analyze statelessness correctly
            builder.getNamedAttr("is_stateless", builder.getBoolAttr(false))};
      }
      if (auto* opts = options.AsWhileOptions()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md

    `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
        %0 = "tf.IfRegion"(%arg0) ( {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 02 02:26:39 UTC 2023
    - 96.4K bytes
    - Viewed (0)
  3. 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