| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | Simplifications.cpp | 77 if (llvm::all_of(opMeshAxes, [&mesh](MeshAxis axis) { in matchAndRewrite()
|
| /llvm-project/mlir/include/mlir/Dialect/Mesh/IR/ |
| H A D | MeshOps.h | 94 for (MeshAxis axis : meshAxes) { collectiveProcessGroupSize() local
|
| /llvm-project/mlir/lib/Dialect/Quant/Utils/ |
| H A D | FakeQuantSupport.cpp | 164 for (size_t axis = 0; axis != axisSize; ++axis) { in fakeQuantAttrsToType() local
|
| /llvm-project/mlir/lib/Dialect/Mesh/IR/ |
| H A D | MeshOps.cpp | 138 for (auto axis : axes) { verifyMeshAxes() local
347 for (MeshAxis axis : axesArray) { verify() local
590 for (int64_t axis = 0; axis < operandType.getRank(); ++axis) { verifyGatherOperandAndResultShape() local
608 for (int64_t axis = 0; axis < operandType.getRank(); ++axis) { verifyAllToAllOperandAndResultShape() local
653 for (int64_t axis = 0; axis < operandType.getRank(); ++axis) { verifyScatterOrSliceOperandAndResultShape() local
[all...] |
| /llvm-project/mlir/lib/ExecutionEngine/ |
| H A D | CRunnerUtils.cpp | 111 for (int64_t axis = rank - 1; axis >= 0; --axis) { in memrefCopy() local
|
| /llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ |
| H A D | ShardingInterface.cpp | 235 for (MeshAxis axis : meshAxes) { fillShardingOption() local
424 const SmallVector<MeshAxis> &axis = std::get<1>(it); getShardingAttribute() local
|
| /llvm-project/mlir/lib/Dialect/XeGPU/IR/ |
| H A D | XeGPUOps.cpp | 216 auto axis = getVnniAxis().value(); verify() local
|
| /llvm-project/mlir/lib/Dialect/Tosa/IR/ |
| H A D | TosaOps.cpp | 277 const int64_t axis = getAxisAttr().getInt(); verify() local
566 IntegerAttr axis = adaptor.getProperties().axis; inferReturnTypeComponents() local
629 int32_t axis = prop.axis.getValue().getSExtValue(); inferReturnTypeComponents() local
1246 ReduceInferReturnTypes(ShapeAdaptor operandShape,Type inputType,IntegerAttr axis,SmallVectorImpl<ShapedTypeComponents> & inferredReturnShapes) ReduceInferReturnTypes() argument
[all...] |
| H A D | TosaCanonicalizations.cpp | 394 int32_t axis = concatOp.getAxis(); matchAndRewrite() local
906 auto axis = getAxis(); fold() local
[all...] |
| /llvm-project/mlir/lib/Conversion/TosaToTensor/ |
| H A D | TosaToTensor.cpp | 385 int axis = op.getAxis(); matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | MeshShardingInterfaceImpl.cpp | 192 __anon3ed07fe50f02(MeshAxis axis) createAllReduceForResultWithoutPartialSharding() argument
|
| /llvm-project/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalg.cpp | 1038 reduceMatchAndRewriteHelper(Operation * op,uint64_t axis,PatternRewriter & rewriter) reduceMatchAndRewriteHelper() argument
1823 auto axis = op.getAxis(); matchAndRewrite() local
1962 int axis = argmaxOp.getAxis(); matchAndRewrite() local
[all...] |