| H A D | MeshOps.cpp | 491 auto meshAxes = op.getMeshAxes(); matchAndRewrite() local
510 verifyInGroupDevice(Location loc,StringRef deviceName,ArrayRef<int64_t> device,Operation::operand_range deviceDynamic,ArrayRef<MeshAxis> meshAxes,ArrayRef<int64_t> meshShape) verifyInGroupDevice() argument
578 verifyGatherOperandAndResultShape(Value operand,Value result,int64_t gatherAxis,ArrayRef<MeshAxis> meshAxes,ArrayRef<int64_t> meshShape) verifyGatherOperandAndResultShape() argument
605 verifyAllToAllOperandAndResultShape(Value operand,Value result,int64_t splitAxis,int64_t concatAxis,ArrayRef<MeshAxis> meshAxes,ArrayRef<int64_t> meshShape) verifyAllToAllOperandAndResultShape() argument
650 verifyScatterOrSliceOperandAndResultShape(Value operand,Value result,int64_t tensorAxis,ArrayRef<MeshAxis> meshAxes,ArrayRef<int64_t> meshShape) verifyScatterOrSliceOperandAndResultShape() argument
687 sliceResultType(Type operandType,MeshOp mesh,ArrayRef<MeshAxis> meshAxes,int64_t sliceAxis) sliceResultType() argument
744 build(OpBuilder & odsBuilder,OperationState & odsState,Value input,StringRef mesh,ArrayRef<MeshAxis> meshAxes,ReductionKind reduction) build() argument
774 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
1026 auto meshAxes = getMeshAxes(); verifySymbolUses() local
[all...] |