Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 67 for NamedAttribute (0.17 sec)

  1. 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)
  2. 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)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_op_interfaces.h

    //===----------------------------------------------------------------------===//
    
    struct ContractionFusion {
      explicit ContractionFusion(
          StringRef output_kernel, ArrayRef<int> additional_arguments = {},
          ArrayRef<NamedAttribute> additional_attributes = {})
          : output_kernel(output_kernel.str()),
            additional_arguments(additional_arguments.begin(),
                                 additional_arguments.end()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 03 19:26:14 UTC 2023
    - 6.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/passes/manipulate_model_attr.cc

      auto entry_func_attr =
          func_op->getAttrOfType<DictionaryAttr>(kTfEntryFunctionAttr);
      if (!entry_func_attr) return;
    
      auto entry_func_attrs = SmallVector<NamedAttribute>(entry_func_attr.begin(),
                                                          entry_func_attr.end());
    
      MLIRContext* ctx = func_op.getContext();
      for (auto& named_attr : entry_func_attrs) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 26 01:13:26 UTC 2023
    - 2.1K bytes
    - Viewed (0)
  5. 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)
  6. tensorflow/compiler/mlir/lite/flatbuffer_operator.h

        // NOLINTNEXTLINE
        Location loc, llvm::SmallVectorImpl<mlir::NamedAttribute> *attributes);
    
    // TODO(zichuanwei@): Populate Builtin_options_2 manual for now, should automate
    // these in the future
    void BuiltinOptions2ToAttributes(
        tflite::BuiltinOptions2Union op_union, mlir::Builder builder,
        llvm::SmallVectorImpl<mlir::NamedAttribute> &attributes);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 21:00:09 UTC 2024
    - 11.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc

    LogicalResult SmuggleOp(Operation* op, PatternRewriter& rewriter) {
      auto call_target =
          rewriter.getNamedAttr("call_target_name", op->getName().getIdentifier());
      SmallVector<NamedAttribute> attrs{op->getAttrs()};
      attrs.push_back(call_target);
      auto custom_call = rewriter.create<mlir::stablehlo::CustomCallOp>(
          op->getLoc(), op->getResultTypes(), op->getOperands(), attrs);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 19 01:14:04 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc

          temp[i] = vec[i].AsInt64();
        }
        return temp;
      }
    
      llvm::SmallVector<mlir::NamedAttribute, 4> ReadAttr(const flexbuffers::Map& m,
                                                          Builder* builder,
                                                          std::string op_name) {
        llvm::SmallVector<mlir::NamedAttribute, 4> attrs;
        const auto& keys = m.Keys();
        for (size_t i = 0; i < keys.size(); ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 06:08:43 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc

              versions.bad_consumers().data(),
              versions.bad_consumers().data() + versions.bad_consumers().size())));
      module->setAttr("tf.versions",
                      b.getDictionaryAttr(llvm::ArrayRef<mlir::NamedAttribute>(
                          {producer, min_consumer, bad_consumers})));
    }
    
    mlir::LogicalResult ExtractTfVersions(mlir::ModuleOp module,
                                          VersionDef* versions) {
      versions->Clear();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc

        // Create the TF op
        OperationState op_state(op.getLoc(), tf_op_full_name);
        op_state.addOperands(op.getOperands());
        op_state.addTypes(op.getResultTypes());
    
        SmallVector<NamedAttribute, 2> attrs;
        std::string parsed_op_name;
        tensorflow::NodeDef node_def;
        if (failed(ParseCustomOption(op.getCustomOption().getValue(), op.getLoc(),
                                     parsed_op_name, attrs, node_def))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
Back to top