Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 213 for attr_ (0.44 sec)

  1. 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)
  2. 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)
  3. 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)
  4. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

        llvm::SmallVectorImpl<mlir::NamedAttribute> &output) {
      for (mlir::NamedAttribute attr : attrs) {
        if (attr.getName().strref().contains(
                mlir::tfg::TFGraphDialect::getDeviceAttrKey())) {
          tensorflow::DeviceNameUtils::ParsedName parsed_name;
          if (!tensorflow::DeviceNameUtils::ParseFullName(
                  mlir::cast<mlir::StringAttr>(attr.getValue()).getValue().str(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.cc

      for (it = attrs.begin(); it != attrs.end(); it++) {
        if (count == i) {
          return it->first.length();
        }
        count++;
      }
      return -1;
    }
    
    void TF_OperationGetAttrName(TF_Operation* oper, int i, char* output,
                                 TF_Status* status) {
      auto attrs = oper->node.attrs();
      int count = 0;
      AttrValueMap::const_iterator it;
      for (it = attrs.begin(); it != attrs.end(); it++) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  6. platforms/core-execution/workers/src/test/groovy/org/gradle/workers/internal/IsolatableSerializerRegistryTest.groovy

        }
    
        def "can serialize/deserialize isolated Attribute values"() {
            Attribute attr1 = Attribute.of("foo", String.class)
            Attribute attr2 = Attribute.of("bar", SomeType.class)
            Isolatable<?>[] isolatables = [isolatableFactory.isolate(attr1), isolatableFactory.isolate(attr2)]
    
            when:
            serialize(isolatables)
    
            and:
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Wed May 08 14:30:36 UTC 2024
    - 11.5K bytes
    - Viewed (0)
  7. 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)
  8. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

    }
    
    Status ConvertAttribute(const mlir::BoolAttr& attr, AttrValue* value) {
      value->set_b(attr.getValue());
      return absl::OkStatus();
    }
    
    Status ConvertAttribute(const mlir::IntegerAttr& attr, AttrValue* value) {
      value->set_i(attr.getInt());
      return absl::OkStatus();
    }
    
    Status ConvertAttribute(const mlir::FloatAttr& attr, AttrValue* value) {
      value->set_f(attr.getValueAsDouble());
      return absl::OkStatus();
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc

        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))) {
          return failure();
        }
        if (parsed_op_name != tf_op_name) {
          return op.emitOpError(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

        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.
        Attribute epsilon = rewriter.getF32FloatAttr(0);
        attrs.push_back(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
Back to top