- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ranked_type (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
RankedTensorType DropFirstDimension(Type type) { RankedTensorType ranked_type = mlir::dyn_cast<RankedTensorType>(type); if (!ranked_type) return {}; llvm::ArrayRef<int64_t> dims_except_first = ranked_type.getShape().drop_front(); return tensorflow::GetTypeFromTFTensorShape(dims_except_first, ranked_type.getElementType()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// Returned attribute has integer elements of the given width. static Attribute ConvertShapeToAttr(Type input_ty, int out_width) { auto ranked_ty = input_ty.dyn_cast<RankedTensorType>(); if (!ranked_ty || !ranked_ty.hasStaticShape()) return {}; auto shape = ranked_ty.getShape(); int rank = shape.size(); SmallVector<APInt, 4> dimensions; dimensions.reserve(rank); for (int i = 0; i < rank; ++i)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
limit = rewriter.getIntegerAttr(element_type, type.getShape()[dimension]); delta = rewriter.getIntegerAttr(element_type, 1); } else { return failure(); } auto range_type = RankedTensorType::get({type.getShape()[dimension]}, element_type); Value start_op = rewriter.create<TF::ConstOp>(iota_op.getLoc(), start);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)