Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 58 for NamedAttribute (0.28 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc

    // of the index path attribute.
    SmallVector<NamedAttribute> ReplaceBoundInputAttrWithIndexPathAttr(
        const ArrayRef<NamedAttribute> arg_attrs, const StringRef index_path,
        Builder& builder) {
      // Keep all other attributes except the tf_saved_model.bound_input attribute,
      // as we are replacing it with tf_saved_model.index_path.
      SmallVector<NamedAttribute> new_arg_attrs;
      for (auto arg_attr : arg_attrs) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/common/lift_as_function_call.td

    // (NamedAttributeList (NamedAttr<"transpose_a"> $transpose_a))
    def NamedAttributeList : NativeCodeCall<"llvm::SmallVector<NamedAttribute>{$0...}">;
    
    // Creates a NamedAttribute given its name and value. Essentially creates
    // a pair: {attribute_name, attribute_value}.
    class NamedAttr<string attr_name> :
        NativeCodeCall<"NamedAttribute{$_builder.getStringAttr(\"" # attr_name # "\"), $0}">;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 25 00:32:20 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_composite_to_tfl_custom.cc

            NamedAttribute num_layers_attr(
                num_layers_str,
                IntegerAttr::get(IntegerType::get(fn.getContext(), /*width=*/32),
                                 num_layers));
    
            // current_layer_index Composite Attribute.
            mlir::StringAttr current_layer_str =
                builder.getStringAttr("layer_index");
            NamedAttribute current_layer_attr(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h

                                             NamedAttribute named_attr) override;
      LogicalResult verifyRegionResultAttribute(Operation *op,
                                                unsigned region_index,
                                                unsigned result_index,
                                                NamedAttribute named_attr) override;
      LogicalResult verifyOperationAttribute(Operation *op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 18 03:21:34 UTC 2024
    - 5.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

      }
      return mlir::success();
    }
    
    static mlir::LogicalResult ReformatOpAttributes(
        mlir::MLIRContext *context, llvm::ArrayRef<mlir::NamedAttribute> attrs,
        llvm::SmallVectorImpl<mlir::NamedAttribute> &output) {
      for (mlir::NamedAttribute attr : attrs) {
        if (attr.getName().strref().contains(
                mlir::tfg::TFGraphDialect::getDeviceAttrKey())) {
          tensorflow::DeviceNameUtils::ParsedName parsed_name;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/add_dump_tensor_op.cc

            log_dir_path_(std::move(log_dir_path)) {}
    
     private:
      SmallVector<NamedAttribute> CreateDumpAttributes(
          PatternRewriter &rewriter, const StringRef folder_name,
          const StringRef file_name, const bool enabled, const StringRef func_name,
          const StringRef node_name) const {
        SmallVector<NamedAttribute> dump_attributes{
            rewriter.getNamedAttr("log_dir_path",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 22:55:22 UTC 2024
    - 13K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

        // contraction, with two additions: the list of ops which have been fused
        // together; epsilon (only with FusedBatchNorm).
        std::vector<NamedAttribute> attrs = contraction->getAttrs();
        ArrayAttr fused_ops_attr = ArrayAttr::get(context, fused_ops);
        attrs.push_back(
            NamedAttribute(StringAttr::get(context, "fused_ops"), fused_ops_attr));
        // Epsilon is used only in fusions with the FusedBatchNorm op, so we zero it
        // here.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h

    void CopyAttributes(Operation *from, Operation *to, Predicate P) {
      for (const NamedAttribute &attr : from->getAttrs())
        if (P(attr)) to->setAttr(attr.getName(), attr.getValue());
    }
    
    // Copies attributes whose name begins with an _ from `from` to `to`.
    inline void CopyUnderscoredAttributes(Operation *from, Operation *to) {
      CopyAttributes(from, to, [](const NamedAttribute &attr) {
        return attr.getName().strref().front() == '_';
      });
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 19:47:48 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc

                                  const ArrayRef<NamedAttribute> attributes,
                                  const ArrayRef<Operation*> ops) {
      // A map to find which operation an attribute belongs to.
      // The key for this map uses the entire NamedAttribute object, i.e. the
      // {attribute_name, attribute_value} pair.
      llvm::SmallDenseMap<NamedAttribute, Operation*> attr_to_op_map;
      for (Operation* op : ops) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc

      addOperations<
    #define GET_OP_LIST
    #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc.inc"
          >();
    }
    
    static LogicalResult VerifyIndexPath(Operation *op, NamedAttribute named_attr) {
      auto attr = mlir::dyn_cast<ArrayAttr>(named_attr.getValue());
      if (!attr) {
        return op->emitError() << "'" << kTfSavedModelIndexPathAttr
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 23.5K bytes
    - Viewed (0)
Back to top