Searched refs:use_chain_endings (Results 1 – 1 of 1) sorted by relevance
3559 const ConstHloInstructionSet& use_chain_endings) { in AreUsersElementwise() argument3561 return use_chain_endings.count(user) || in AreUsersElementwise()3563 AreUsersElementwise(user, use_chain_endings)); in AreUsersElementwise()3571 const ConstHloInstructionSet& use_chain_endings) { in NumInputsInvolveInOnlyElementwiseOps() argument3576 AreUsersElementwise(parameter, use_chain_endings); in NumInputsInvolveInOnlyElementwiseOps()3614 ConstHloInstructionSet use_chain_endings; in IsUnrollingColumnReductionBeneficial() local3616 use_chain_endings.insert(fused_root); in IsUnrollingColumnReductionBeneficial()3629 use_chain_endings.insert(instr); in IsUnrollingColumnReductionBeneficial()3635 unnested_hlo, input_shape, use_chain_endings); in IsUnrollingColumnReductionBeneficial()