Searched defs:sliceAxis (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | Transforms.cpp | 116 int64_t sliceAxis = op.getSliceAxis().getSExtValue(); in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Mesh/IR/ |
| H A D | MeshOps.cpp | 688 sliceResultType(Type operandType,MeshOp mesh,ArrayRef<MeshAxis> meshAxes,int64_t sliceAxis) sliceResultType() argument
775 build(OpBuilder & odsBuilder,OperationState & odsState,Value input,MeshOp mesh,ArrayRef<MeshAxis> meshAxes,int64_t sliceAxis) build() argument
783 build(OpBuilder & odsBuilder,OperationState & odsState,Type resultType,Value input,StringRef mesh,ArrayRef<MeshAxis> meshAxes,int64_t sliceAxis) build() argument
|