Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc

          if (IsQI8Type(result_type) || IsQUI8Type(result_type)) {
            builder->setInsertionPoint(op);
            TFL::QuantizeOp quant_op = builder->create<TFL::QuantizeOp>(
                op->getLoc(), result_type, new_result, TypeAttr::get(result_type));
            new_result = quant_op.getResult();
          }
    
          // Rewire the outputs.
          result.replaceAllUsesWith(new_result);
        }
    
        // Remove the old op.
        op->erase();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

    // the casting, the quantization dimension of the result type needs to be set
    // this new `axis` value.
    TypeAttr CastQuantizedTypeAttrFromExpressedType(Builder builder,
                                                    TypeAttr source, Type target,
                                                    int axis);
    
    // Quantizes the elements in the attribute `real_value` by the quantization
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc

        builder.create<tf_saved_model::GlobalTensorOp>(
            NameLoc::get(builder.getStringAttr(name.str())),
            builder.getStringAttr(name), tensor_attr,
            TypeAttr::get(tensor_attr.getType()), builder.getUnitAttr());
      }
    
      return success();
    }
    
    }  // namespace
    
    LogicalResult LiftVariables(ModuleOp module, Session* session) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 7.9K bytes
    - Viewed (0)
  4. 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)
  5. tensorflow/compiler/mlir/tfr/python/tfr_gen.py

            attr_type, attr_def.name, attr_def.type)
      else:
        attr_names = []
        if arg_def.number_attr:
          attr_names.append(arg_def.number_attr)
        if arg_def.type_attr:
          attr_names.append(arg_def.type_attr)
        if arg_def.type_list_attr:
          attr_names.append(arg_def.type_list_attr)
    
        # TODO(fengliuai): currently we don't support backward type inference, so we
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 27 15:27:03 UTC 2022
    - 55.8K bytes
    - Viewed (0)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top