Home
last modified time | relevance | path

Searched defs:dims (Results 1 – 25 of 69) sorted by relevance

123

/llvm-project/mlir/lib/IR/
H A DTypeUtilities.cpp62 for (auto dims : llvm::zip(shape1, shape2)) { in verifyCompatibleShape() local
104 LogicalResult mlir::verifyCompatibleDims(ArrayRef<int64_t> dims) { in verifyCompatibleDims() argument
157 auto dims = llvm::map_to_vector<8>( verifyCompatibleShapes() local
[all...]
/llvm-project/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h132 : dims(std::forward<Arg>(arg)), elementType(elementType), attr(attr), in dims() function
158 ShapeStorageT dims; variable
/llvm-project/openmp/runtime/test/worksharing/for/
H A Dkmp_doacross_check.c27 struct dim dims; in main() local
/llvm-project/mlir/unittests/Analysis/Presburger/
H A DParserTest.cpp26 unsigned dims, unsigned syms, ArrayRef<SmallVector<int64_t, 4>> ineqs, in makeFACFromConstraints()
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp43 SmallVector<unsigned> dims; in splitReduction() local
257 SmallVector<unsigned> dims; in splitReductionByScaling() local
316 SmallVector<Value> dims = in splitReductionByScaling() local
H A DSpecialize.cpp181 auto dims = *res; in specializeLinalgContractions() local
[all...]
H A DBlockPackMatmul.cpp46 ArrayRef<int64_t> dims) { in validateFullTilesOnDims() argument
/llvm-project/flang/test/Lower/Intrinsics/
H A Dreshape.f9080 integer :: dims(4) = [2,2,2,2] local
/llvm-project/mlir/examples/toy/Ch3/include/toy/
H A DAST.h81 std::vector<int64_t> dims; variable
85 std::vector<int64_t> dims) in LiteralExprAST()
H A DParser.h98 std::vector<int64_t> dims; in parseTensorLiteralExpr() local
/llvm-project/mlir/examples/toy/Ch6/include/toy/
H A DAST.h81 std::vector<int64_t> dims; variable
85 std::vector<int64_t> dims) in LiteralExprAST()
H A DParser.h98 std::vector<int64_t> dims; in parseTensorLiteralExpr() local
/llvm-project/mlir/examples/toy/Ch5/include/toy/
H A DAST.h81 std::vector<int64_t> dims; variable
85 std::vector<int64_t> dims) in LiteralExprAST()
H A DParser.h98 std::vector<int64_t> dims; in parseTensorLiteralExpr() local
/llvm-project/mlir/examples/toy/Ch4/include/toy/
H A DAST.h81 std::vector<int64_t> dims; variable
85 std::vector<int64_t> dims) in LiteralExprAST()
H A DParser.h98 std::vector<int64_t> dims; in parseTensorLiteralExpr() local
/llvm-project/mlir/examples/toy/Ch2/include/toy/
H A DAST.h81 std::vector<int64_t> dims; variable
85 std::vector<int64_t> dims) in LiteralExprAST()
H A DParser.h98 std::vector<int64_t> dims; in parseTensorLiteralExpr() local
/llvm-project/mlir/examples/toy/Ch1/include/toy/
H A DAST.h81 std::vector<int64_t> dims; variable
85 std::vector<int64_t> dims) in LiteralExprAST()
H A DParser.h98 std::vector<int64_t> dims; in parseTensorLiteralExpr() local
/llvm-project/mlir/lib/Dialect/GPU/IR/
H A DInferIntRangeInterfaceImpls.cpp44 static Value valueByDim(KernelDim3 dims, Dimension dim) { in valueByDim()
59 getKnownLaunchAttr(GPUFuncOp func, LaunchDims dims, Dimension dim) { in getKnownLaunchAttr()
/llvm-project/mlir/examples/toy/Ch7/include/toy/
H A DAST.h83 std::vector<int64_t> dims; variable
87 std::vector<int64_t> dims) in LiteralExprAST()
H A DParser.h113 std::vector<int64_t> dims; in parseTensorLiteralExpr() local
/llvm-project/mlir/lib/Interfaces/
H A DInferTypeOpInterface.cpp180 SmallVector<int64_t> dims; dump() local
/llvm-project/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgElementwiseFusion.cpp258 [&dims](linalg::LinalgOp op) { in runOnOperation() local

123