- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 112 for ShapedType (0.34 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
for (auto s : arr) { if (s != iota) return false; ++iota; } return true; } PermutationAndShape GetPermutationAndTransposedShape( llvm::ArrayRef<int64_t> permutation_array, ShapedType input_type, ConversionPatternRewriter& rewriter) { assert(permutation_array.size() == input_type.getRank()); llvm::SmallVector<int64_t> transposed_shape(permutation_array.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
if (expected_out_row_dim != ShapedType::kDynamic && out_row_dim != ShapedType::kDynamic && out_row_dim != expected_out_row_dim) return op.emitOpError() << "found invalid output dimension on row, expected " << expected_out_row_dim << " but got " << out_row_dim; if (expected_out_col_dim != ShapedType::kDynamic && out_col_dim != ShapedType::kDynamic &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.h
// not in the op registry will be ignored. If the `ignore_unregistered_attrs` // argument is not set to true, _output_shapes attribute is added to nodes with // ShapedType for the leading values with ShapedType in the results of the // nodes. Set it to true if the returned NodeDef will be executed by the linked // TF Eager runtime. absl::StatusOr<std::unique_ptr<NodeDef>> ConvertTFDialectOpToNodeDef(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/UniformSupport.cc
return isa<FloatType>(input_type); } 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/quantization/tensorflow/passes/quantize_composite_functions.cc
return failure(); } ShapedType tensor_qtype = mlir::cast<ShapedType>(q_op.getResult().getType()); Attribute tensor_proto_attr = Quantize(attr, tensor_qtype); if (!tensor_proto_attr) { return failure(); } Type storage_type = mlir::cast<QuantizedType>(tensor_qtype.getElementType()) .getStorageType(); ShapedType new_type = tensor_qtype.clone(storage_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/QuantizeUtils.cc
if (!quantDenseAttr) { return nullptr; } // Cast from an expressed-type-based type to storage-type-based type, // preserving the sparse shape (i.e. tensor<4xf32> -> tensor<4xi8>). ShapedType newSparseType = mlir::dyn_cast_or_null<ShapedType>( quantizedElementType.castExpressedToStorageType( realSparseAttr.getType())); if (!newSparseType) { return nullptr; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/shlo_simplify.td
include "mlir/IR/BuiltinAttributes.td" include "mlir/IR/CommonAttrConstraints.td" include "mlir/IR/CommonTypeConstraints.td" def CloneF32ElementsAttrWithOnes : NativeCodeCall<"DenseElementsAttr::get($0.getType().cast<ShapedType>(), (float)1.0)">; def NotConstant : Constraint< CPred<"$0.isa<BlockArgument>() || !llvm::isa<stablehlo::ConstantOp>($0.getDefiningOp())">, "Is not a constant.">; def : Pat<(StableHLO_DivOp $l,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 03:05:20 UTC 2024 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc
if (op->getNumResults() > 0 && isa<ShapedType>(op->getResult(0).getType())) { // Use rhs operand to detect types for dynamic range quantizable ops. Value value_for_deducing_op_type = (dyn_cast_or_null<DynamicRangeQuantizedOpInterface>(op)) ? op->getOperand(1) : op->getResult(0); ShapedType value_shaped_type = mlir::dyn_cast_or_null<ShapedType>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
if (!reshape_op) return failure(); auto reshape_type = mlir::cast<ShapedType>(reshape_op.getOutput().getType()); if (!reshape_type.hasStaticShape()) return failure(); ArrayRef<int64_t> reshape_shape = reshape_type.getShape(); auto input_type = mlir::cast<ShapedType>(op.getInput().getType()); auto output_type = mlir::cast<ShapedType>(op.getOutput().getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
} // Here we only add the shapes for the leading values with ShapedType, // assuming values with non-ShapedType are put at the end of the result. if (!ignore_unregistered_attrs && inst->getNumResults() > 0) { auto values = inst->getResults(); auto begin = values.begin(); auto end = values.begin(); while (end != values.end() && mlir::isa<mlir::ShapedType>((*end).getType())) end++; if (begin != end) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0)