- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ranked_type (0.12 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
Type& shard_type) { RankedTensorType ranked_type = mlir::dyn_cast<RankedTensorType>(full_type); if (!ranked_type) return context_op->emitOpError() << "A map_outside_compilation op's input and output types must be " "ranked tensors."; ArrayRef<int64_t> in_shape = ranked_type.getShape(); if (in_shape.empty() || in_shape[0] < 0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
// also populate the shape info to the op result. The shape of the // tensorlist is inferred from `num_elements` and `element_shape`. auto ranked_type = element_shape.getType().dyn_cast<RankedTensorType>(); DenseIntElementsAttr dense_elem_attr; if ((ranked_type && ranked_type.getRank() == 0) || !matchPattern(element_shape, m_Constant(&dense_elem_attr))) { // If no constant is spotted, just forward the operand.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
// to expand at the given `axis`. Type InferExpandDimsType(Type ty, int64_t axis, Builder *builder) { auto ranked_ty = mlir::dyn_cast<RankedTensorType>(ty); // Unranked type. if (!ranked_ty) return ty; auto shape = llvm::to_vector<4>(ranked_ty.getShape()); if (axis < 0) axis += ranked_ty.getRank() + 1; shape.insert(shape.begin() + axis, 1); return tensorflow::GetTypeFromTFTensorShape(shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0)