Searched refs:fused_instruction_count (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_fusion_test.cc | 82 EXPECT_EQ(4, fusion_instruction->fused_instruction_count()); in TEST_F()
129 EXPECT_EQ(8, fusion_instruction->fused_instruction_count()); in TEST_F()
205 EXPECT_EQ(6, fusion_instruction1->fused_instruction_count()) in TEST_F()
214 EXPECT_EQ(5, fusion_instruction2->fused_instruction_count()) in TEST_F()
274 EXPECT_EQ(4, fusion1->fused_instruction_count()); in TEST_F()
275 EXPECT_EQ(4, fusion2->fused_instruction_count()); in TEST_F()
332 EXPECT_EQ(3, fusion_inst->fused_instruction_count()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fusion_merger_test.cc | 91 EXPECT_EQ(4, operand0->fused_instruction_count()); in TEST_F()
95 EXPECT_EQ(7, operand1->fused_instruction_count()); in TEST_F()
99 EXPECT_EQ(7, operand2->fused_instruction_count()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 275 std::vector<HloOpcode> fused_opcodes(root->fused_instruction_count()); in RunFusionAndCheckOpcodesWereFused()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 726 EXPECT_EQ(entry_comp->root_instruction()->fused_instruction_count(), 6); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 1467 int64 fused_instruction_count() const;
|
| D | hlo_instructions.h | 774 int64 fused_instruction_count() const;
|
| D | hlo_instruction.cc | 3338 int64 HloInstruction::fused_instruction_count() const { in fused_instruction_count() function in xla::HloInstruction
3339 return Cast<HloFusionInstruction>(this)->fused_instruction_count(); in fused_instruction_count()
|
| D | hlo_instructions.cc | 1364 int64 HloFusionInstruction::fused_instruction_count() const { in fused_instruction_count() function in xla::HloFusionInstruction
|