Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 73 for getElementDtype (0.21 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc

        SmallVector<int64_t, 4> new_shape(permutation.size());
        for (size_t i = 0; i < permutation.size(); ++i)
          new_shape[i] = shape[permutation[i]];
    
        return RankedTensorType::get(new_shape, ranked_type.getElementType());
      }
    
      return type;
    }
    
    bool AreCancellablePermutations(DenseIntElementsAttr perm0,
                                    DenseIntElementsAttr perm1) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/common/ir/UniformSupport.cc

    }
    
    ExpressedToQuantizedConverter ExpressedToQuantizedConverter::forInputType(
        Type input_type) {
      if (isa<TensorType, VectorType>(input_type)) {
        Type element_type = cast<ShapedType>(input_type).getElementType();
        if (!isQuantizablePrimitiveType(element_type))
          return ExpressedToQuantizedConverter{input_type, nullptr};
        return ExpressedToQuantizedConverter{input_type, element_type};
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/experimental/tac/common/utils.h

    // Returns true if it is a shaped type of f32 elements.
    inline bool IsF32ShapedType(Type t) {
      if (auto shaped_type = mlir::dyn_cast_or_null<ShapedType>(t)) {
        return shaped_type.getElementType().isF32();
      }
      return false;
    }
    
    // Return true when the given element_type is QI8.
    inline bool IsQI8Type(Type t) {
      auto quantized_type = quant::QuantizedType::getQuantizedElementType(t);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc

      auto original = mlir::dyn_cast<ShapedType>(original_ty);
    
      // Both types must be shaped types.
      if (!original || !updated) return false;
    
      // Element types must match.
      if (original.getElementType() != updated.getElementType()) return false;
    
      // If the updated type doesn't have a rank, then it can't be a more refined
      // type.
      if (!updated.hasRank()) return false;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 9.7K bytes
    - Viewed (0)
  5. platforms/core-configuration/model-core/src/main/java/org/gradle/api/internal/provider/CollectionPropertyInternal.java

    import java.util.Collection;
    
    public interface CollectionPropertyInternal<T, C extends Collection<T>> extends PropertyInternal<C>, HasMultipleValues<T>, CollectionProviderInternal<T, C> {
        @Override
        Class<T> getElementType();
    
        /**
         * Adds an element to the property value.
         *
         * <p>
         * When invoked on a property with no value, this method first sets the value
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Thu May 23 18:32:13 UTC 2024
    - 3.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/hoist_replicate_invariant_resource_writes.cc

    // similar. Move to common utils.
    bool IsResourceType(Type type) {
      return type.isa<TF::ResourceType>() ||
             (type.isa<TensorType>() &&
              type.cast<TensorType>().getElementType().isa<TF::ResourceType>());
    }
    
    SmallVector<Value> GetAccessedResources(Operation& op) {
      SmallVector<Value, 4> accessed_resources;
      for (auto operand : op.getOperands()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 5.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_tensor_helper.h

    // given `rank`.
    inline bool IsOfRankedFloatTensorType(RankedTensorType type, int rank) {
      return type && type.getRank() == rank &&
             mlir::isa<FloatType>(type.getElementType());
    }
    
    // Returns true if the given `value` has the specified rank or has unranked
    // type.
    inline bool IsOfRankOrUnranked(Value value, int64_t rank) {
      RankedTensorType type = GetRankedTensorTypeForOperand(value);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.8K bytes
    - Viewed (0)
  8. platforms/core-runtime/internal-instrumentation-processor/src/main/java/org/gradle/internal/instrumentation/processor/codegen/groovy/SignatureTree.java

                // In the end, match the vararg parameter, if it is there:
                varargParameter.map(parameterInfo -> Stream.of(new ParameterMatchEntry(parameterInfo.getParameterType().getElementType(), VARARG))).orElseGet(Stream::empty)
            ).flatMap(Function.identity()).collect(Collectors.toList());
        }
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Mon Oct 02 15:44:14 UTC 2023
    - 5.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc

                                                         res_attr);
      return success();
    }
    
    static LogicalResult FoldDivOp(stablehlo::DivOp op, PatternRewriter& rewriter) {
      auto etype = op.getType().getElementType();
      if (etype.isa<FloatType>()) {
        return FoldDivOpInternal<APFloat>(op, rewriter);
      }
      if (etype.isa<IntegerType>()) {
        return FoldDivOpInternal<APInt>(op, rewriter);
      }
      return failure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 06:11:55 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_arith_ops_folder.h

      const int identity =
          (std::is_same<OpT, MulOp>::value || std::is_same<OpT, DivOp>::value ||
           std::is_same<OpT, RealDivOp>::value)
              ? 1
              : 0;
    
      Type element_ty = lhs_type.getElementType();
      Attribute identity_attr;
      if (auto ty = mlir::dyn_cast<FloatType>(element_ty)) {
        identity_attr = FloatAttr::get(ty, static_cast<double>(identity));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.3K bytes
    - Viewed (0)
Back to top