Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 77 for attr_ (0.15 sec)

  1. tensorflow/c/experimental/ops/gen/cpp/views/attr_view.cc

      string c_base_type = attr_.base_type();
      if (attr_.base_type() == "type") {
        c_base_type = "DataType";
      } else if (attr_.base_type() == "shape") {
        c_base_type = "const PartialTensorShape";
      }
    
      // ... and wrap in a Span<> if it's a list.
      if (attr_.is_list()) {
        return absl::Substitute("absl::Span<$0>", c_base_type);
      } else {
        return c_base_type;
      }
    
      return attr_.full_type();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 03 07:02:00 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc

      }
    
      auto attrs = attr_.getAttrs();
    
      if (failed(HasIntegerArrayWithSize(&func_, attrs, "pool_size", 2))) {
        return failure();
      }
    
      if (failed(HasIntegerArrayWithSize(&func_, attrs, "strides", 2))) {
        return failure();
      }
    
      // Retrieves padding.
      auto padding = mlir::dyn_cast_or_null<StringAttr>(attrs.get("padding"));
      if (!padding) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/nms_utils.cc

      }
    
      auto attrs = attr_.getAttrs();
      if (failed(HasIntAttr(func_, attrs, "max_detections")) ||
          failed(HasIntAttr(func_, attrs, "max_classes_per_detection")) ||
          failed(HasIntAttr(func_, attrs, "num_classes")) ||
          failed(HasFloatAttr(func_, attrs, "nms_score_threshold")) ||
          failed(HasFloatAttr(func_, attrs, "nms_iou_threshold")) ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/utils/convert_attr.cc

          // will not use this representation.
          mlir::NamedAttrList attrs;
          for (const auto& func_attr : value.func().attr()) {
            TF_ASSIGN_OR_RETURN(auto attr,
                                ConvertAttributeValue(func_attr.second, builder));
            attrs.push_back(builder->getNamedAttr(func_attr.first, attr));
          }
          auto func_attrs = builder->getDictionaryAttr(attrs);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc

          DeviceNameUtils::ParsedName device;
          if (DeviceNameUtils::ParseFullName(device_name, &device) &&
              device.has_job && device.job == jobType) {
            for (const auto& attr : node->attrs()) {
              auto attr_key = attr.first;
              auto attr_value = attr.second;
              if (attr_key == attrKey &&
                  attr_value.value_case() == AttrValue::kType &&
                  attr_value.type() == DT_RESOURCE) {
                return true;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 12:22:33 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td

                                /*filter_width*/(GetI32At<1> (GetAsVectorAttr<"kernel_size"> $attrs)),
                                /*padding*/(GetAvgPoolOpPadAttr $old_val),
                                /*stride_h*/(GetI32At<0> (GetAsVectorAttr<"stride"> $attrs)),
                                /*stride_w*/(GetI32At<1> (GetAsVectorAttr<"stride"> $attrs)),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 23:16:05 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h

      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)
  8. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.cc

        return mlir::failure();
    
      int64_t num_results = 0;
      if (succeeded(parser.parseOptionalColon())) {
        mlir::IntegerAttr attr;
        mlir::NamedAttrList attrs;
        if (failed(parser.parseAttribute(attr, "num_results", attrs)))
          return mlir::failure();
        num_results = attr.getValue().getSExtValue();
      }
      if (num_results < 0) return mlir::failure();
    
      llvm::SmallVector<mlir::Type, 4> operand_types;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.h

    bool EnsureAttribute(const DictionaryAttr& composite_attributes,
                         const std::string& attr_name, AttrType* out_attr) {
      Attribute attr = composite_attributes.get(attr_name);
      if (!mlir::isa_and_nonnull<AttrType>(attr)) {
        return false;
      }
      if (AttrType content = mlir::dyn_cast<AttrType>(attr)) {
        *out_attr = content;
        return true;
      } else {
        return false;
      }
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 18:33:05 UTC 2024
    - 3.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/node_matchers_test.cc

          "to be 0 but found 1");
    }
    
    TEST(NodeMatchers, Attrs) {
      Scope root = Scope::NewRootScope().ExitOnError();
      Output enter = ops::internal::Enter(
          root.WithOpName("enter"),
          ops::Placeholder(root.WithOpName("data"), DT_FLOAT), "frame_name",
          ops::internal::Enter::Attrs{}.IsConstant(true));
      EXPECT_THAT(enter.node(), NodeWith(Attr("is_constant", true)));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 14:43:57 UTC 2024
    - 9.1K bytes
    - Viewed (0)
Back to top