Home
last modified time | relevance | path

Searched refs:ReusesOperandElements (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.cc69 ReusesOperandElements(consumer, operand_index)) { in ShouldFuseInexpensiveChecks()
Dfusion_merger.cc250 return user->ReusesOperandElements(operand_index); in HandleFusion()
/external/tensorflow/tensorflow/compiler/xla/service/
Dinstruction_fusion.h144 bool ReusesOperandElements(const HloInstruction* consumer,
Dinstruction_fusion.cc724 bool InstructionFusion::ReusesOperandElements(const HloInstruction* consumer, in ReusesOperandElements() function in xla::InstructionFusion
731 bool reuses = consumer->ReusesOperandElements(operand_index); in ReusesOperandElements()
Dhlo_instruction_test.cc1966 EXPECT_FALSE(root->ReusesOperandElements(0)); in TEST_F()
1989 EXPECT_TRUE(root->ReusesOperandElements(0)); in TEST_F()
2002 EXPECT_FALSE(root->ReusesOperandElements(0)); in TEST_F()
2020 EXPECT_FALSE(root->ReusesOperandElements(0)); in TEST_F()
2021 EXPECT_FALSE(root->ReusesOperandElements(1)); in TEST_F()
Dhlo_instruction.h1500 bool ReusesOperandElements(int64 i) const { in ReusesOperandElements() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc98 ReusesOperandElements(consumer, operand_index)) { in ShouldFuse()