- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for ranked_type (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
} // Shuffle ranked tensor dimensions according to the permutation. Type ShuffleRankedTensorType(Type type, ArrayRef<int64_t> permutation) { if (auto ranked_type = mlir::dyn_cast<RankedTensorType>(type)) { ArrayRef<int64_t> shape = ranked_type.getShape(); assert(permutation.size() == shape.size()); SmallVector<int64_t, 4> new_shape(permutation.size()); for (size_t i = 0; i < permutation.size(); ++i)
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/tensorflow/ir/tf_ops_tensor_helper.cc
} else { return UnrankedTensorType::get(builder->getI1Type()); } } auto ranked_type = mlir::dyn_cast<RankedTensorType>(result_type); if (!ranked_type) return UnrankedTensorType::get(builder->getI1Type()); return RankedTensorType::get(ranked_type.getShape(), builder->getI1Type()); } Type InferReductionOpType(Value input, Value reduction_indices,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/validators.cc
if (dim != 1) return false; } return true; } bool IsDimensionsDegenerateExceptLastOne(TypedAttr val) { if (auto ranked_type = mlir::dyn_cast<RankedTensorType>(val.getType())) { return IsDimensionsDegenerateExceptLastOne(ranked_type.getShape()); } return false; } } // namespace TFL
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc
// dimensions. bool IsBounded(Type ty) { auto ranked_ty = mlir::dyn_cast<RankedTensorType>(ty); if (!ranked_ty) return false; if (ranked_ty.hasStaticShape()) return true; auto encoding = mlir::dyn_cast_or_null<TypeExtensionsAttr>(ranked_ty.getEncoding()); if (!encoding) return false; for (int i = 0; i < ranked_ty.getRank(); ++i) { if (ranked_ty.isDynamicDim(i) &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 9.7K bytes - Viewed (0)