- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 73 for getElementDtype (0.21 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)