Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.cc69 ReusesOperandElements(consumer, operand_index)) { in ShouldFuseInexpensiveChecks()
Dfusion_merger.cc310 return user->ReusesOperandElements(operand_index); in HandleFusion()
/external/tensorflow/tensorflow/compiler/xla/service/
Dinstruction_fusion.h268 bool ReusesOperandElements(const HloInstruction* consumer,
Dinstruction_fusion.cc1043 bool reuses = instruction->ReusesOperandElements(i); in ReusedOperandsOf()
1053 bool InstructionFusion::ReusesOperandElements(const HloInstruction* consumer, in ReusesOperandElements() function in xla::InstructionFusion
Dhlo_instruction_test.cc2138 EXPECT_FALSE(root->ReusesOperandElements(0)); in TEST_F()
2161 EXPECT_TRUE(root->ReusesOperandElements(0)); in TEST_F()
2174 EXPECT_FALSE(root->ReusesOperandElements(0)); in TEST_F()
2192 EXPECT_FALSE(root->ReusesOperandElements(0)); in TEST_F()
2193 EXPECT_FALSE(root->ReusesOperandElements(1)); in TEST_F()
Dhlo_instruction.h1668 bool ReusesOperandElements(int64_t i) const;
Dhlo_instruction.cc4012 bool HloInstruction::ReusesOperandElements(int64_t i) const { in ReusesOperandElements() function in xla::HloInstruction
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc95 ReusesOperandElements(consumer, operand_index)) { in ShouldFuse()