Searched defs:splitMeshAxis (Results 1 – 1 of 1) sorted by relevance
111 MeshAxis splitMeshAxis) { in targetShardingInSplitLastAxis() argument 135 int64_t splitTensorAxis, MeshAxis splitMeshAxis) { in splitLastAxisInResharding() argument 263 unsplitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshShardingAttr sourceSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard,MeshOp mesh,int64_t splitTensorAxis,MeshAxis splitMeshAxis) unsplitLastAxisInResharding() argument