Home
last modified time | relevance | path

Searched refs:MakeScopedInstructionProfiler (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmemset_thunk.cc28 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
37 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dcopy_thunk.cc38 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
59 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dcudnn_batchnorm_thunk.cc110 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
182 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
266 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dsequential_thunk.cc39 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dtuple_thunk.cc38 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dfor_thunk.cc48 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dhlo_execution_profiler.h67 std::unique_ptr<ScopedInstructionProfiler> MakeScopedInstructionProfiler(
Dconvolution_thunk.cc58 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dwhile_thunk.cc56 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dconditional_thunk.cc62 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Doutfeed_thunk.cc36 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dhlo_execution_profiler.cc111 HloExecutionProfiler::MakeScopedInstructionProfiler( in MakeScopedInstructionProfiler() function in xla::gpu::HloExecutionProfiler
Dkernel_thunk.cc101 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dinfeed_thunk.cc35 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dfft_thunk.cc120 auto op_profiler = profiler->MakeScopedInstructionProfiler(hlo_instruction()); in ExecuteOnStream()
Dgemm_thunk.cc468 auto op_profiler = profiler->MakeScopedInstructionProfiler( in ExecuteOnStream()