Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for UpdateDataFormat (0.32 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h

                                    DenseIntElementsAttr perm1);
    
    // Default implementation of `LayoutSensitiveInterface::UpdateDataFormat` for
    // layout sensitive operations that do not have any additional layout dependent
    // attributes besides `data_format` string.
    template <typename Op>
    LogicalResult UpdateDataFormat(StringRef data_format, Op *op) {
      auto perm = GetDataFormatPermutation(op->getDataFormat(), data_format);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 5.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

                  "found "
               << feature_dim << " and " << bias_len << ", respectively";
      }
      return success();
    }
    
    LogicalResult BiasAddOp::UpdateDataFormat(StringRef data_format) {
      return ::mlir::TF::UpdateDataFormat(data_format, this);
    }
    
    StringRef BiasAddOp::GetOptimalLayout(const RuntimeDevices& devices) {
      // Keep current data format if no GPUs are available or if explicit placement
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_op_interfaces.td

        >,
        InterfaceMethod<
          [{Updates operation attributes and operands to account for the updated
            data format. If data format is not supported, must return failure.}],
          "LogicalResult", "UpdateDataFormat", (ins "StringRef":$data_format)
        >,
      ];
    
      let verify = [{
        return VerifyLayoutSensitiveInterface($_op);
      }];
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Nov 30 19:07:07 UTC 2022
    - 6.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc

          return DenseIntElementsAttr::get(perm_ty, permutation);
        };
    
        // Change operation data format.
        if (failed(layout_sensitive_interface.UpdateDataFormat(target_data_format)))
          return;
    
        // Permute arguments into the target data format.
        builder.setInsertionPoint(op);
        auto arg_perm = builder.create<ConstOp>(loc, perm_attr(args_permutation));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        SmallVector<unsigned, 4> GetLayoutDependentArgs() { return {0}; }
        SmallVector<unsigned, 4> GetLayoutDependentResults() { return {0}; }
        StringRef GetOptimalLayout(const RuntimeDevices& devices);
        LogicalResult UpdateDataFormat(StringRef data_format);
      }];
    
      let hasVerifier = 1;
    }
    
    def TF_BiasAddGradOp : TF_Op<"BiasAddGrad", [Pure]> {
      let summary = [{
    The backward operation for "BiasAdd" on the "bias" tensor.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top