Home
last modified time | relevance | path

Searched refs:XLA_SCOPED_LOGGING_TIMER (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_configuration_ops.cc112 XLA_SCOPED_LOGGING_TIMER("ConfigureDistributedTpuOp"); in Compute()
147 XLA_SCOPED_LOGGING_TIMER("WaitForDistributedTpuOp"); in Compute()
237 XLA_SCOPED_LOGGING_TIMER("ShutdownDistributedTpuOp"); in Compute()
253 XLA_SCOPED_LOGGING_TIMER("InitializeHostForDistributedTpuOp"); in Compute()
372 XLA_SCOPED_LOGGING_TIMER("SetGlobalTPUArrayOp"); in Compute()
388 XLA_SCOPED_LOGGING_TIMER("DisconnectDistributedTpuChipsOp"); in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Damdgpu_compiler.cc131 XLA_SCOPED_LOGGING_TIMER( in CompileTargetBinary()
Dnvptx_compiler.cc327 XLA_SCOPED_LOGGING_TIMER( in CompileTargetBinary()
344 XLA_SCOPED_LOGGING_TIMER("NVPTXCompiler::CompileGpuAsmOrGetCachedResult"); in CompileGpuAsmOrGetCachedResult()
Dgpu_compiler.cc493 XLA_SCOPED_LOGGING_TIMER("GpuCompiler::RunHloPasses"); in RunHloPasses()
609 XLA_SCOPED_LOGGING_TIMER("GpuCompiler::RunBackend - IR emission"); in CompileModuleToLlvmIrImpl()
688 XLA_SCOPED_LOGGING_TIMER( in CompileToTargetBinary()
858 XLA_SCOPED_LOGGING_TIMER("GpuCompiler::RunBackend"); in RunBackend()
Dgpu_conv_algorithm_picker.cc383 XLA_SCOPED_LOGGING_TIMER(absl::StrCat( in PickBestAlgorithmNoCacheCuda()
657 XLA_SCOPED_LOGGING_TIMER(absl::StrCat( in PickBestAlgorithmNoCacheRocm()
837 XLA_SCOPED_LOGGING_TIMER("GpuConvAlgorithmPicker"); in Run()
Dbuffer_comparator.cc689 XLA_SCOPED_LOGGING_TIMER("BufferComparator::CompareEqual"); in CompareEqualParameterized()
Dgemm_algorithm_picker.cc316 XLA_SCOPED_LOGGING_TIMER("GemmAlgorithmPicker"); in Run()
Dgpu_executable.cc419 XLA_SCOPED_LOGGING_TIMER(absl::StrCat( in ExecuteAsyncOnStreamImpl()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_pass_pipeline.cc165 XLA_SCOPED_LOGGING_TIMER(absl::StrCat("HLO pass: ", pass->name())); in RunPassesInternal()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/
Dgpu_backend_lib.cc525 XLA_SCOPED_LOGGING_TIMER("Compile module " + module->getName().str()); in CompileToPtx()
912 XLA_SCOPED_LOGGING_TIMER("Compile module " + module->getName().str()); in CompileToHsaco()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc523 XLA_SCOPED_LOGGING_TIMER("CpuCompiler - Running LLVM verifier"); in VerifyLlvmModule()
658 XLA_SCOPED_LOGGING_TIMER( in RunBackend()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.cc409 XLA_SCOPED_LOGGING_TIMER("Compilation of XLA executable"); in CompileImpl()
/external/tensorflow/tensorflow/compiler/xla/
Dutil.h79 #define XLA_SCOPED_LOGGING_TIMER(label) \ macro