Home
last modified time | relevance | path

Searched refs:SetCyclesTakenBy (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_execution_profiler.cc71 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/
Dhlo_execution_profile_test.cc64 execution_profile.SetCyclesTakenBy(add_instruction, add_cycles); in TEST_F()
65 execution_profile.SetCyclesTakenBy(dot_instruction, dot_cycles); in TEST_F()
Dhlo_execution_profile.cc134 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
Dhlo_execution_profile.h112 void SetCyclesTakenBy(const HloInstruction* hlo, uint64 cycles_taken);
115 void SetCyclesTakenBy(size_t index, uint64 cycles_taken);