Home
last modified time | relevance | path

Searched refs:resultShape (Results 1 – 9 of 9) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/
DTraits.cpp35 SmallVectorImpl<int64_t> &resultShape) { in getBroadcastedShape() argument
44 resultShape.clear(); in getBroadcastedShape()
46 std::copy(shape1.begin(), shape1.end(), std::back_inserter(resultShape)); in getBroadcastedShape()
48 std::copy(shape2.begin(), shape2.end(), std::back_inserter(resultShape)); in getBroadcastedShape()
53 auto iR = resultShape.rbegin(); in getBroadcastedShape()
80 resultShape.clear(); in getBroadcastedShape()
149 SmallVector<int64_t, 4> resultShape; in getBroadcastedType() local
150 if (!getBroadcastedShape(getShape(type1), getShape(type2), resultShape)) in getBroadcastedType()
155 return VectorType::get(resultShape, elementType); in getBroadcastedType()
157 return RankedTensorType::get(resultShape, elementType); in getBroadcastedType()
[all …]
/external/llvm-project/mlir/lib/Dialect/Shape/IR/
DShape.cpp330 SmallVector<int64_t, 6> resultShape; in fold() local
333 if (!OpTrait::util::getBroadcastedShape(lhsShape, rhsShape, resultShape)) in fold()
336 return builder.getIndexTensorAttr(resultShape); in fold()
350 SmallVector<int64_t, 6> resultShape; in fold() local
351 resultShape.append(lhsShape.begin(), lhsShape.end()); in fold()
352 resultShape.append(rhsShape.begin(), rhsShape.end()); in fold()
354 return builder.getIndexTensorAttr(resultShape); in fold()
452 SmallVector<int64_t, 6> resultShape; in fold() local
/external/llvm-project/mlir/include/mlir/Dialect/
DTraits.h48 SmallVectorImpl<int64_t> &resultShape);
/external/llvm-project/flang/lib/Evaluate/
Dfold-implementation.h186 ConstantSubscripts resultShape; in ApplySubscripts() local
191 resultShape.push_back(static_cast<ConstantSubscript>(ss.size())); in ApplySubscripts()
205 CHECK(k < GetRank(resultShape)); in ApplySubscripts()
209 if (++ssAt[k] == resultShape[k]) { in ApplySubscripts()
226 CHECK(k == GetRank(resultShape)); in ApplySubscripts()
229 return Constant<T>{array.LEN(), std::move(values), std::move(resultShape)}; in ApplySubscripts()
232 std::move(resultShape)}; in ApplySubscripts()
234 return Constant<T>{std::move(values), std::move(resultShape)}; in ApplySubscripts()
Dcheck-expression.cpp320 } else if (auto resultShape{GetShape(context, folded)}) { in NonPointerInitializationExpr() local
322 *resultShape, "initialized object", in NonPointerInitializationExpr()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_op_base.td604 SmallVector<int64_t, 4> resultShape;
607 y.getType().cast<ShapedType>().getShape(), resultShape)) {
612 resultType = RankedTensorType::get(resultShape, $_builder.getI1Type());
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorOps.cpp2576 auto resultShape = resultVectorType.getShape(); in verifyVectorShapeCast() local
2582 resultShape.begin(), resultShape.end(), 1LL, std::multiplies<int64_t>{}); in verifyVectorShapeCast()
2590 if (!isValidShapeCast(sourceShape, resultShape)) in verifyVectorShapeCast()
2593 if (!isValidShapeCast(resultShape, sourceShape)) in verifyVectorShapeCast()
2958 auto resultShape = resultType.getShape(); in verify() local
2962 if (attrValue < 0 || attrValue > resultShape[it.index()]) in verify()
DVectorTransforms.cpp493 auto resultShape = resultType.getShape(); in getVectorElementwiseOpUnrollState() local
500 indexMap.reserve(resultShape.size()); in getVectorElementwiseOpUnrollState()
501 for (unsigned i = 0; i < resultShape.size(); ++i) in getVectorElementwiseOpUnrollState()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc726 auto resultShape = resultType.getShape(); in Verify() local
727 if (resultShape != llvm::makeArrayRef(expectedShape)) { in Verify()
730 llvm::make_range(resultShape.begin(), resultShape.end()), in Verify()