Searched refs:use_chain_endings (Results 1 – 1 of 1) sorted by relevance
5192 const absl::flat_hash_set<mlir::Operation*>& use_chain_endings) { in AreUsersElementwise() argument5196 return use_chain_endings.count(user) || in AreUsersElementwise()5198 AreUsersElementwise(user->getResult(0), use_chain_endings)); in AreUsersElementwise()5206 const absl::flat_hash_set<mlir::Operation*>& use_chain_endings) { in NumInputsInvolveInOnlyElementwiseOps() argument5211 AreUsersElementwise(parameter, use_chain_endings); in NumInputsInvolveInOnlyElementwiseOps()5250 absl::flat_hash_set<mlir::Operation*> use_chain_endings; in IsUnrollingColumnReductionBeneficial() local5254 use_chain_endings.insert(fusion_results[0].getDefiningOp()); in IsUnrollingColumnReductionBeneficial()5267 use_chain_endings.insert(result.getDefiningOp()); in IsUnrollingColumnReductionBeneficial()5273 use_chain_endings); in IsUnrollingColumnReductionBeneficial()