Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 29 for attrName (0.17 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc

      return true;
    }
    
    bool GetI32VectorFromDenseI64CompositeAttr(
        const DictionaryAttr& composite_attrs, const std::string& attr_name,
        std::vector<int32_t>* out_vec) {
      DenseIntElementsAttr attr;
      if (!EnsureAttribute<DenseIntElementsAttr>(composite_attrs, attr_name,
                                                 &attr)) {
        return false;
      }
    
      return DenseI64AttrToI32Vector(attr, out_vec);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 18:33:05 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

      for (auto& attr : func->getAttrs()) {
        StringRef attr_name = attr.getName().getValue();
        if (attr_name.starts_with("tf_saved_model.")) {
          func->removeAttr(attr_name);
        }
      }
    
      auto iface = cast<FunctionOpInterface>(func.getOperation());
      for (int i = 0; i < func.getNumArguments(); ++i) {
        for (auto& attr : iface.getArgAttrs(i)) {
          const StringAttr& attr_name = attr.getName();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc

            [&](const tensorflow::NameRangeMap& arg_ranges,
                const tensorflow::protobuf::RepeatedPtrField<
                    tensorflow::OpDef::ArgDef>& args,
                llvm::StringRef attr_name) {
              std::vector<int32_t> values;
              values.reserve(args.size());
              for (const auto& arg : args) {
                auto range = arg_ranges.at(arg.name());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc

        // TODO(b/146937733): Don't use <void> here.
        llvm::StringRef attr_name = mlir::OpTrait::AttrSizedOperandSegments<
            void>::getOperandSegmentSizeAttr();
        attrs_to_ignore.insert(attr_name.data());
      }
    
      if (inst->hasTrait<mlir::OpTrait::AttrSizedResultSegments>()) {
        // TODO(b/146937733): Don't use <void> here.
        llvm::StringRef attr_name = mlir::OpTrait::AttrSizedResultSegments<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc

      for (auto& attr : op->getAttrs()) {
        llvm::StringRef attr_name = attr.getName().getValue();
        if (attribute_getter_map.find(attr_name.str()) !=
            attribute_getter_map.end()) {
          auto attr_val =
              (attribute_getter_map[attr_name.str()])(rewriter, identifier_to_attr);
          attrs.push_back(rewriter.getNamedAttr(attr_name, attr_val));
        }
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 18.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc

      void runOnOperation() override;
    };
    
    LogicalResult SetOnceModuleAttribute(StringRef attr_name,
                                         IntegerAttr attr_value, Operation* op,
                                         ModuleOp& module) {
      const auto ex_attr_value = module->getAttrOfType<IntegerAttr>(attr_name);
      if (ex_attr_value == nullptr) {
        module->setAttr(attr_name, attr_value);
        return success();
      }
      if (ex_attr_value == attr_value) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_unified_experimental.cc

      }
      tsl::Set_TF_Status_from_Status(s, tracing_op->SetOpName(op_name));
    }
    
    void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name,
                                  TF_DataType value, TF_Status* s) {
      Status status =
          unwrap(op)->SetAttrType(attr_name, static_cast<DataType>(value));
      TF_SetStatus(s, static_cast<TF_Code>(status.code()),
                   absl::StatusMessageAsCStr(status));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 10:15:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tfr/passes/decompose.cc

          } else {
            auto attr_name = compose_func.getArgAttrOfType<StringAttr>(
                arg.index(), kAttrArgumentNameAttr);
            auto attribute = op->getAttr(attr_name.getValue());
            if (!attribute) {
              attribute =
                  compose_func.getArgAttr(arg.index(), kAttrArgumentDefaultAttr);
            }
            if (!attribute && attr_name.getValue() == "out_type") {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc

        if (operand.index() >= call_op.getNumOperands()) {
          auto attr_name = signature.getArgAttrOfType<StringAttr>(
              operand.index(), kAttrArgumentNameAttr);
          auto attr_value =
              signature.getArgAttr(operand.index(), kAttrArgumentDefaultAttr);
          arg_attrs->push_back(
              rewriter.getNamedAttr(attr_name.getValue(), attr_value));
          continue;
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          }
        }
    
        auto check_i64_attr =
            [&](const std::string& attr_name) -> mlir::LogicalResult {
          if (!backend_config.contains(attr_name)) {
            return op.emitOpError()
                   << "Missing " << attr_name << " attribute in backend_config";
          }
          auto attr = backend_config.getAs<IntegerAttr>(attr_name);
          if (!attr || !attr.getType().isInteger(64)) {
            return op.emitOpError()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
Back to top