Searched refs:callee_root (Results 1 – 2 of 2) sorted by relevance
1132 auto* callee_root = user->to_apply()->root_instruction(); in CanShareOperandBufferWithUser() local1134 absl::c_find_if(uses, [callee_root](const HloUse& use) { in CanShareOperandBufferWithUser()1135 return use.instruction == callee_root && in CanShareOperandBufferWithUser()1136 callee_root->IsElementwiseOnOperand(use.operand_number); in CanShareOperandBufferWithUser()
812 auto* callee_root = user->to_apply()->root_instruction(); in CanShareOperandBufferWithUser() local813 return param_uses.size() == 1 && param_uses[0].first == callee_root && in CanShareOperandBufferWithUser()814 callee_root->IsElementwiseOnOperand(param_uses[0].second); in CanShareOperandBufferWithUser()