Searched refs:makeStridedLinearLayoutMap (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_gpu.cc | 124 makeStridedLinearLayoutMap(llvm::None, in matchAndRewrite()
|
| D | hlo_legalize_to_lhlo.cc | 355 makeStridedLinearLayoutMap(dynamic_layout, in InsertDynamicMemrefCastOp()
|
| D | legalize_to_linalg.cc | 647 makeStridedLinearLayoutMap(new_strides, operand_offset, in InsertReshapeIfNecessary()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | BuiltinTypes.cpp | 682 AffineMap mlir::makeStridedLinearLayoutMap(ArrayRef<int64_t> strides, in makeStridedLinearLayoutMap() function in mlir
783 return MemRefType::Builder(t).setAffineMaps(makeStridedLinearLayoutMap( in eraseStridedLayout()
|
| /external/llvm-project/mlir/lib/Parser/ |
| D | TypeParser.cpp | 253 map = makeStridedLinearLayoutMap(strides, offset, state.context); in parseMemRefType()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | BuiltinTypes.h | 780 AffineMap makeStridedLinearLayoutMap(ArrayRef<int64_t> strides, int64_t offset,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 67 makeStridedLinearLayoutMap(strides, /*offset=*/0, builder.getContext())}; in GetPermutationIfAvailable()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 780 makeStridedLinearLayoutMap(intStrides, intOffset, type.getContext()); in computeReshapeCollapsedType()
1038 .setAffineMaps(makeStridedLinearLayoutMap( in build()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/IR/ |
| D | Ops.cpp | 3067 makeStridedLinearLayoutMap(targetStrides, targetOffset, in inferResultType()
3277 auto reducedMap = makeStridedLinearLayoutMap(keepStrides, originalOffset, c); in isRankReducedType()
4019 makeStridedLinearLayoutMap(strides, offset, memRefType.getContext()); in inferTransposeResultType()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 2206 makeStridedLinearLayoutMap(resStrides, resOffset, aT.getContext())); in getCastCompatibleMemRefType()
|