Searched refs:resultShape (Results 1 – 9 of 9) sorted by relevance
/external/llvm-project/mlir/lib/Dialect/ |
D | Traits.cpp | 35 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/ |
D | Shape.cpp | 330 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/ |
D | Traits.h | 48 SmallVectorImpl<int64_t> &resultShape);
|
/external/llvm-project/flang/lib/Evaluate/ |
D | fold-implementation.h | 186 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()
|
D | check-expression.cpp | 320 } else if (auto resultShape{GetShape(context, folded)}) { in NonPointerInitializationExpr() local 322 *resultShape, "initialized object", in NonPointerInitializationExpr()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
D | tf_op_base.td | 604 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/ |
D | VectorOps.cpp | 2576 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()
|
D | VectorTransforms.cpp | 493 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/ |
D | hlo_ops.cc | 726 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()
|