Searched refs:IsElementwiseOnOperand (Results 1 – 6 of 6) sorted by relevance
65 if (!user->IsElementwiseOnOperand(use_index) && in AreTransitiveUsesElementwiseOrTuple()1136 callee_root->IsElementwiseOnOperand(use.operand_number); in CanShareOperandBufferWithUser()1143 return user->IsElementwiseOnOperand(user->operand_index(operand)); in CanShareOperandBufferWithUser()
814 callee_root->IsElementwiseOnOperand(param_uses[0].second); in CanShareOperandBufferWithUser()821 return user->IsElementwiseOnOperand(user->operand_index(operand)); in CanShareOperandBufferWithUser()
1015 EXPECT_TRUE(add->IsElementwiseOnOperand(i)); in TEST_F()1080 EXPECT_FALSE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()1082 EXPECT_TRUE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()1120 EXPECT_FALSE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()1122 EXPECT_TRUE(fusion->IsElementwiseOnOperand(operand_idx)); in TEST_F()
1235 bool IsElementwiseOnOperand(int64 operand_idx) const;
47 return instruction->IsElementwiseOnOperand(operand_index); in IsInstructionElementwiseOnOperand()
2857 bool HloInstruction::IsElementwiseOnOperand(int64 operand_idx) const { in IsElementwiseOnOperand() function in xla::HloInstruction