Searched refs:EvaluateEmittedInstructions (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | fusion_node_indexing_evaluation_test.cc | 64 int64 EvaluateEmittedInstructions(const HloInstruction* producer, in EvaluateEmittedInstructions() function in xla::InstructionFusionForTesting
74 return fusion_node_evaluations_.at(consumer).EvaluateEmittedInstructions( in EvaluateEmittedInstructions()
112 EXPECT_EQ(instruction_fusion.EvaluateEmittedInstructions(slice2, fusion), 1); in TEST_F()
154 EXPECT_EQ(instruction_fusion.EvaluateEmittedInstructions(slice2_1, fusion), in TEST_F()
160 EXPECT_EQ(instruction_fusion.EvaluateEmittedInstructions(add1, fusion), 2); in TEST_F()
165 EXPECT_EQ(instruction_fusion.EvaluateEmittedInstructions(slice1_0, fusion), in TEST_F()
171 EXPECT_EQ(instruction_fusion.EvaluateEmittedInstructions(slice1_1, fusion), in TEST_F()
177 EXPECT_EQ(instruction_fusion.EvaluateEmittedInstructions(add0, fusion), 4); in TEST_F()
210 EXPECT_EQ(instruction_fusion.EvaluateEmittedInstructions(add0, fusion), 4); in TEST_F()
|
| D | fusion_node_indexing_evaluation.cc | 84 int64 emitted_instructions = EvaluateEmittedInstructions(producer); in CodeDuplicationTooHigh()
101 int64 FusionNodeIndexingEvaluation::EvaluateEmittedInstructions( in EvaluateEmittedInstructions() function in xla::FusionNodeIndexingEvaluation
|
| D | fusion_node_indexing_evaluation.h | 42 int64 EvaluateEmittedInstructions(const HloInstruction* producer) const;
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 156 producer, eval_consumer.EvaluateEmittedInstructions(producer)); in IsFusedIrEmitterInefficient()
|