Home
last modified time | relevance | path

Searched refs:PtxOptsFromDebugOptions (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_asm_opts_util.h26 stream_executor::GpuAsmOpts PtxOptsFromDebugOptions(
Dgpu_asm_opts_util.cc28 stream_executor::GpuAsmOpts PtxOptsFromDebugOptions( in PtxOptsFromDebugOptions() function
Dbuffer_comparator.cc635 PtxOptsFromDebugOptions(config.debug_options())); in DeviceCompare()
Dgemm_algorithm_picker.cc72 stream, allocator, PtxOptsFromDebugOptions(debug_options), in CreateRedzoneAllocator()
Dnvptx_compiler.cc412 PtxOptsFromDebugOptions(hlo_module_config.debug_options()); in CompileGpuAsmOrGetCachedResult()
Dgpu_conv_algorithm_picker.cc491 PtxOptsFromDebugOptions(hlo_module_config.debug_options()), in AutotuneOneConvRunner()
667 PtxOptsFromDebugOptions(hlo_module_config.debug_options()), in PickBestAlgorithmNoCacheCuda()
Djitrt_custom_calls.cc1336 RunCholesky(xla::gpu::PtxOptsFromDebugOptions(*debug_options), in operator ()()
1475 a_data, result_data, temp_data, PtxOptsFromDebugOptions(*debug_options), in operator ()()
Dir_emitter_unnested.cc1262 PtxOptsFromDebugOptions(hlo_module_config_.debug_options()), a_buffer, in EmitCholeskyThunk()
1472 PtxOptsFromDebugOptions(hlo_module_config_.debug_options()), in EmitTriangularSolveCustomCall()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc147 xla::gpu::PtxOptsFromDebugOptions(config.debug_options()); in GetGpuBinaryBlob()