Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 57 for qtype_attr (0.17 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model_ops.td

        sense of `tf.TensorShape` compatibility. And the element types must match.
      }];
    
      let arguments = (ins
        StrAttr:$sym_name,
        OptionalAttr<ElementsAttr>:$value,
        TypeAttr:$type,
        UnitAttr:$is_mutable
      );
    
      let hasVerifier = 1;
    }
    
    def TfSavedModel_SessionInitializerOp: TfSavedModel_Op<"session_initializer"> {
      let summary = "Initializes TensorFlow session state.";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:49:35 UTC 2023
    - 7.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/utils.h

          values.begin(), values.end(),
          [](const APInt& element_value) { return element_value.isNegative(); });
    }
    
    // Utility function to get the offset between two dense attribute values.
    inline TypedAttr GetOffSet(Attribute begin, Attribute end) {
      auto begin_values = begin.cast<DenseElementsAttr>().getValues<int32_t>();
      auto end_values = end.cast<DenseElementsAttr>().getValues<int32_t>();
    
      SmallVector<int32_t> offsets;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/optimize.cc

        return f.isExactlyValue(value);
      });
    }
    
    // Returns true if `value` is compile-time constant and its splat value equals
    // to `raw_value`.
    template <typename T>
    bool IsConstantValueOf(mlir::TypedAttr value, T raw_value) {
      auto element_type = mlir::cast<ShapedType>(value.getType()).getElementType();
    
      if (mlir::isa<FloatType>(element_type)) {
        return FloatValueEquals(value, raw_value);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td

              "mlir::OperandElementTypeIterator(values.end())};",
      [{
        ArrayAttr::get($_ctxt,
        [&]() {
          llvm::SmallVector<Attribute, 4> ret;
          for (auto t : $_self)
            ret.push_back(TypeAttr::get(t));
          return ret;
        }())
      }]
    >;
    
    // A derived attribute that returns the shapes of the tensors in the actual
    // value pack that corresponds to the `idx`-th ODS-declared variadic operand.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 30.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

      if (node.IsArg()) {
        if (dtype == DT_RESOURCE) {
          const AttrValue* dtype_attr = node.attrs().Find("_handle_dtypes");
          const AttrValue* shape_attr = node.attrs().Find("_handle_shapes");
          if (dtype_attr && shape_attr) {
            if (dtype_attr->list().type().empty()) {
              return errors::InvalidArgument(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

          // Here TArgs types do not include types of the first two parameters,
          // i.e. the convolution input and the filter. TArgs are parameters for
          // the extras like the bias etc.
          auto attr = TypeAttr::get(getElementTypeOrSelf(contraction.getType()));
          SmallVector<Attribute, 4> targs_values(operands.size() - 2, attr);
    
          ArrayAttr targs_attr = ArrayAttr::get(context, targs_values);
          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)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

                res_types, res_attrs)))
          return failure();
    
        // Update the function type which has excluded the control args.
        func->setAttr("function_type", TypeAttr::get(rewriter.getFunctionType(
                                           arg_types, res_types)));
    
        // Update arg/result attributes.
        func.setAllArgAttrs(arg_attrs);
        func.setAllResultAttrs(res_attrs);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc

            auto min_max = GetMinMaxValuesForArgument(func_name, i);
            // The input min/max or mean/std are not specified, then skip.
            if (!min_max.first.has_value() || !min_max.second.has_value()) return;
    
            TypeAttr params = quant::GetQuantizedTypeAttr(
                builder, input_type, builder.getF64FloatAttr(min_max.first.value()),
                builder.getF64FloatAttr(min_max.second.value()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc

            auto min_max = GetMinMaxValuesForArgument(func_name, i);
            // The input min/max or mean/std are not specified, then skip.
            if (!min_max.first.has_value() || !min_max.second.has_value()) return;
    
            TypeAttr params = quant::GetQuantizedTypeAttr(
                builder, input_type, builder.getF64FloatAttr(min_max.first.value()),
                builder.getF64FloatAttr(min_max.second.value()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

          return false;
        }
      }
      return true;
    }
    
    // Returns the tensor type created from the `shape_attr` and `type_attr`
    // attributes.
    Type GetType(Attribute shape_attr, Attribute type_attr) {
      auto shape = mlir::cast<tf_type::ShapeAttr>(shape_attr);
      auto type = mlir::cast<TypeAttr>(type_attr);
      if (shape.hasRank())
        return tensorflow::GetTypeFromTFTensorShape(shape.getShape(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
Back to top