Searched defs:shardOp (Results 1 – 4 of 4) sorted by relevance
113 if (!shardOp) in getMeshShardingAttr() local 102 auto shardOp = llvm::dyn_cast<mesh::ShardOp>(user); getMeshShardingAttr() local 118 auto shardOp = llvm::dyn_cast<mesh::ShardOp>(user); getMeshShardingAttr() local 126 ShardOp shardOp = llvm::dyn_cast<ShardOp>(user); getMeshShardingAttr() local 145 if (ShardOp shardOp = val.getDefiningOp<ShardOp>()) getMeshShardingAttr() local [all...]
175 ShardOp shardOp = llvm::dyn_cast_or_null<ShardOp>(operand.getDefiningOp()); getReshardingRquirementKind() local 193 ShardOp shardOp = llvm::dyn_cast<ShardOp>(user); getReshardingRquirementKind() local
537 ShardOp shardOp = llvm::dyn_cast<ShardOp>(useOp); shardedBlockArgumentTypes() local 587 ShardOp shardOp = llvm::cast<ShardOp>(definingOp); getOperandShardings() local 608 ShardOp shardOp = llvm::cast<ShardOp>(userOp); getResultShardings() local 615 spmdizeOperation(ShardOp shardOp,IRMapping & spmdizationMap,SymbolTableCollection & symbolTableCollection,OpBuilder & builder) spmdizeOperation() argument 643 ShardOp shardOp = llvm::dyn_cast<ShardOp>(op); spmdizeOperation() local [all...]
189 ShardOp shardOp = dyn_cast<ShardOp>(operandOp); maybeInsertTargetShardingAnnotation() local 230 ShardOp shardOp = dyn_cast_or_null<ShardOp>(operandSrcOp); maybeInsertSourceShardingAnnotation() local [all...]