Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 38 for attr_value (0.21 sec)

  1. tensorflow/cc/framework/cc_op_gen_util.cc

    }
    
    string PrintAttrValue(const string& op, const AttrValue& attr_value) {
      switch (attr_value.value_case()) {
        case AttrValue::kS:
          return PrintString(attr_value.s());
        case AttrValue::kI:
          return strings::StrCat(attr_value.i());
        case AttrValue::kF: {
          const float f = attr_value.f();
          if (std::isinf(f)) {
            return strings::StrCat(f < 0.0f ? "-" : "+",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 25K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/node_matchers.cc

    }
    
    std::pair<string, AttrValue> impl::AttrLiteralHelper(
        const std::pair<string, bool>& bool_attr) {
      AttrValue attr_value;
      attr_value.set_b(bool_attr.second);
      return {bool_attr.first, attr_value};
    }
    
    std::pair<string, AttrValue> impl::AttrLiteralHelper(
        const std::pair<string, absl::Span<const int>>& int_list_attr) {
      AttrValue attr_value;
      AttrValue::ListValue* list = attr_value.mutable_list();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 03 16:15:20 UTC 2022
    - 16.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_cluster_util.cc

      for (const auto& name_attr_pair : n.attrs()) {
        const AttrValue& attr_value = name_attr_pair.second;
        if (attr_value.value_case() == AttrValue::kFunc) {
          result.push_back(attr_value.func());
        } else if (attr_value.value_case() == AttrValue::kList) {
          result.insert(result.end(), attr_value.list().func().begin(),
                        attr_value.list().func().end());
        }
      }
    
      return result;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

          *list->add_tensor() = tensor;
        } else if (auto attr = mlir::dyn_cast<mlir::FlatSymbolRefAttr>(a)) {
          AttrValue attr_val;
          TF_RETURN_IF_ERROR(ConvertAttribute(attr, &attr_val));
          *list->add_func() = attr_val.func();
        } else if (auto attr = mlir::dyn_cast<mlir::TypeAttr>(a)) {
          AttrValue attr_val;
          // For type attributes, we only propagate the element type.
          mlir::Type elt_type = attr.getValue();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc

        for (const auto& name_and_value : node_def.attr()) {
          const std::string& attr_name = name_and_value.first;
          const tensorflow::AttrValue& attr_value = name_and_value.second;
          absl::StatusOr<Attribute> mlir_attr =
              tensorflow::ConvertAttributeValue(attr_value, &builder);
          if (!mlir_attr.ok()) {
            return emitError(loc, mlir_attr.status().message());
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc

                                         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) {
        return success();
      }
    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/compiler/jit/encapsulate_util.cc

    Status AppendToListAttr(Node* n, const string& attr_name, const string& value) {
      std::vector<T> attr_value;
      Status s = GetNodeAttr(n->attrs(), attr_name, &attr_value);
      if (!s.ok() && s.code() != error::NOT_FOUND) {
        return s;
      }
    
      n->ClearAttr(attr_name);
      attr_value.push_back(value);
      n->AddAttr(attr_name, attr_value);
      return absl::OkStatus();
    }
    
    // Replaces attribute value.
    template <typename T>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.5K bytes
    - Viewed (0)
  8. tensorflow/c/c_api_function.cc

                                      const void* proto, size_t proto_len,
                                      TF_Status* status) {
      tensorflow::AttrValue attr_value;
      if (!attr_value.ParseFromArray(proto, proto_len)) {
        status->status = InvalidArgument(
            "Unparseable AttrValue proto passed to "
            "TF_FunctionSetAttrValueProto");
        return;
      }
    
      auto fdef_or = func->record->mutable_fdef();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 13.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

      }
      return true;
    }
    
    bool HasSymbolRefAttr(Operation* op) {
      for (const auto& attr : op->getAttrs()) {
        Attribute attr_value = attr.getValue();
        if (mlir::isa<SymbolRefAttr>(attr_value)) {
          return true;
        } else if (auto array_attr = mlir::dyn_cast<ArrayAttr>(attr_value)) {
          if (!array_attr.empty() &&
              mlir::isa<SymbolRefAttr>(*array_attr.begin())) {
            return true;
          }
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
  10. tensorflow/c/eager/c_api.cc

                                 const void* proto, size_t proto_len,
                                 TF_Status* status) {
      tensorflow::AttrValue attr_value;
      if (!attr_value.ParseFromArray(proto, proto_len)) {
        status->status =
            tensorflow::errors::InvalidArgument("Unparseable AttrValue proto");
        return;
      }
      if (op == nullptr) {
        status->status = tensorflow::errors::InvalidArgument(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
Back to top