Searched refs:SetCyclesTakenBy (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | hlo_execution_profiler.cc | 71 profile_->SetCyclesTakenBy( in FinishExecution() 95 profile_->SetCyclesTakenBy( in FinishHloComputation() 110 profile_->SetCyclesTakenBy(index, GetCyclesTaken(&timers_, sub_streams_, in FinishHloInstruction()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_execution_profile_test.cc | 64 execution_profile.SetCyclesTakenBy(add_instruction, add_cycles); in TEST_F() 65 execution_profile.SetCyclesTakenBy(dot_instruction, dot_cycles); in TEST_F()
|
D | hlo_execution_profile.cc | 134 void HloExecutionProfile::SetCyclesTakenBy(const HloInstruction* hlo, in SetCyclesTakenBy() function in xla::HloExecutionProfile 136 SetCyclesTakenBy(hlo_profile_index_map_.GetProfileIndexFor(*hlo), in SetCyclesTakenBy() 140 void HloExecutionProfile::SetCyclesTakenBy(size_t index, uint64 cycles_taken) { in SetCyclesTakenBy() function in xla::HloExecutionProfile
|
D | hlo_execution_profile.h | 112 void SetCyclesTakenBy(const HloInstruction* hlo, uint64 cycles_taken); 115 void SetCyclesTakenBy(size_t index, uint64 cycles_taken);
|