Home
last modified time | relevance | path

Searched refs:PtxOptsFromConfig (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.h79 se::GpuAsmOpts PtxOptsFromConfig(const HloModuleConfig& hlo_module_config);
Dbuffer_comparator.cc586 PtxOptsFromConfig(config)); in DeviceCompare()
Dgemm_algorithm_picker.cc71 stream, allocator, PtxOptsFromConfig(hlo_module_config), in DoUncachedGemmAutotune()
Dnvptx_compiler.cc372 auto ptxas_config = PtxOptsFromConfig(hlo_module_config); in CompileGpuAsmOrGetCachedResult()
Dgpu_conv_algorithm_picker.cc404 stream, allocator, PtxOptsFromConfig(hlo_module_config)); in PickBestAlgorithmNoCacheCuda()
464 stream, allocator, PtxOptsFromConfig(hlo_module_config)); in PickBestAlgorithmNoCacheCuda()
Dstream_executor_util.cc226 se::GpuAsmOpts PtxOptsFromConfig(const HloModuleConfig& hlo_module_config) { in PtxOptsFromConfig() function
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc165 auto gpu_asm_opts = xla::gpu::PtxOptsFromConfig(config); in GetGpuBinaryBlob()