- Sort Score
- Result 10 results
- Languages All
Results 141 - 150 of 169 for getElementDtype (0.6 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
Permute<int64_t>(result_type.getShape(), kNchwToNhwcPermutation); const TensorType new_result_type = result_type.cloneWith(new_result_shape, result_type.getElementType()); // Create a new `stablehlo.reduce_window` with all relevant attributes // permutated to match the new operand & result type. auto new_reduce_window_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize_drq.cc
QuantizationUnits& quantizable_ops) const { // Non-float tensors do not need quantization. auto type = mlir::dyn_cast<ShapedType>(op.getType()); if (!type || !type.getElementType().isF32()) return false; Value value = op.getResult(); // Check whether dynamic range quantization can be applied. for (auto& use : value.getUses()) { Operation* user = use.getOwner();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc
llvm::SmallVector<int64_t> shape = {pool.n, h, w, pool.c}; auto op_type = mlir::cast<RankedTensorType>(op->getResult(0).getType()); return RankedTensorType::get(shape, op_type.getElementType()); } StringAttr GetAvgPoolOpPadAttr(Builder& builder, CompositeOp op) { const TorchAvgPoolData pool = GetTorchAvgPoolData(op); if (pool.ph == 0 && pool.pw == 0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 9.2K bytes - Viewed (0) -
platforms/jvm/language-java/src/main/java/org/gradle/api/internal/tasks/compile/incremental/asm/ClassDependenciesVisitor.java
} }); } } protected void maybeAddDependentType(Set<String> types, Type type) { while (type.getSort() == Type.ARRAY) { type = type.getElementType(); } if (type.getSort() != Type.OBJECT) { return; } String name = type.getClassName(); if (typeFilter.test(name)) {
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Tue Feb 27 13:49:15 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc
if (cur_rank != 4 || cur_shape[2] == 1) return failure(); TensorType new_shape = RankedTensorType::get( {cur_shape[0], cur_shape[1], 1, cur_shape[2] * cur_shape[3]}, attr.getElementType()); // Inserts a reshape op. auto shape_spec_type = RankedTensorType::get({cur_rank}, rewriter.getIntegerType(64)); auto new_shape_const_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
permuted_shape.begin() + input_rank - 2); return rewriter.create<TFL::TransposeOp>( op->getLoc(), RankedTensorType::get(permuted_shape, input_type.getElementType()), input, permutation_tensor_op.getResult()); }; if (!can_convert_to_fully_connected(op)) { return false; } Value input_lhs = get_real_input_value(op.getX());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.cc
if (auto attr = inst.getAttrOfType<mlir::ElementsAttr>(attr_id)) { mlir::Attribute rand_val; mlir::Type element_type = attr.getShapedType().getElementType(); if (mlir::isa<mlir::IntegerType>(element_type)) { rand_val = mlir::IntegerAttr::get(element_type, std::rand()); } else if (element_type.isF16() || element_type.isF32() ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 11:51:44 UTC 2024 - 14.1K bytes - Viewed (0) -
subprojects/core/src/main/java/org/gradle/api/internal/collections/AbstractIterationOrderRetainingElementSource.java
} return element; } Element<T> cachingElement(CollectionProviderInternal<T, ? extends Iterable<T>> provider) { final Element<T> element = new Element<>(provider.getElementType(), new ElementsFromCollectionProvider<>(provider), this::doAddRealized); if (provider instanceof ChangingValue) {
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Mar 04 20:04:06 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
elemType.isInteger(64)) return true; // Complex<F<32>> is allowed. if (mlir::isa<ComplexType>(elemType) && mlir::cast<ComplexType>(elemType).getElementType().isF32()) return true; // QUINT8 and UI8 are allowed. if (mlir::isa<TF::Quint8Type>(elemType) || (elemType.isInteger(8) && mlir::cast<IntegerType>(elemType).isUnsigned())) return true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
for (int64_t i = 0; i < permutation_array.size(); ++i) { transposed_shape[i] = input_type.getDimSize(permutation_array[i]); } auto transposed_type = RankedTensorType::get(transposed_shape, input_type.getElementType()); return transposed_type; } // Returns shape of a ranked tensor. // Precondition: output_val's is ranked tensor. // Returns a truncated shape when `truncate` is set to true.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0)