| /external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/ |
| nvptx_backend_lib.h | 25 #include "tensorflow/compiler/xla/service/hlo_module_config.h"
41 const HloModuleConfig& hlo_module_config,
|
| nvptx_backend_lib.cc | 168 const HloModuleConfig& hlo_module_config) {
185 switch (hlo_module_config.debug_options().xla_backend_optimization_level()) {
330 const HloModuleConfig& hlo_module_config,
347 hlo_module_config.debug_options().xla_gpu_ftz());
350 if (hlo_module_config.debug_options().xla_gpu_ftz()) {
375 target_triple, GetSmName(compute_capability), hlo_module_config);
389 hlo_module_config.debug_options().xla_backend_optimization_level();
437 void GPUBackendInit(const HloModuleConfig& hlo_module_config) {
466 llvm_ir::InitializeLLVMCommandLineOptions(hlo_module_config);
484 const HloModuleConfig& hlo_module_config,
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| dot_op_emitter.h | 24 #include "tensorflow/compiler/xla/service/hlo_module_config.h"
67 const HloModuleConfig& hlo_module_config,
|
| dot_op_emitter.cc | 116 const HloModuleConfig& hlo_module_config,
212 const HloModuleConfig& hlo_module_config,
222 hlo_module_config_(hlo_module_config),
857 executable_run_options_value, b, hlo_module_config, local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| ir_emitter_nested.h | 45 IrEmitterNested(const HloModuleConfig& hlo_module_config,
|
| nvptx_compiler.cc | 122 const HloModuleConfig& hlo_module_config) {
131 hlo_module_config.debug_options().xla_gpu_cuda_data_dir();
140 const HloModuleConfig& hlo_module_config) {
143 for (const auto& dir : GetCudaRootCandidates(hlo_module_config)) {
153 string GetLibdeviceDir(const HloModuleConfig& hlo_module_config) {
154 const auto& candidate_dirs = GetCudaRootCandidates(hlo_module_config);
168 hlo_module_config);
541 const HloModuleConfig& hlo_module_config) {
545 for (const string& cuda_root : GetCudaRootCandidates(hlo_module_config)) {
586 if (hlo_module_config.debug_options().xla_gpu_disable_ptxas_optimizations())
[all...] |
| ir_emitter_nested.cc | 38 IrEmitterNested::IrEmitterNested(const HloModuleConfig& hlo_module_config,
41 : IrEmitter(hlo_module_config, ir_emitter_context, /*is_nested=*/true) {
|
| nvptx_compiler.h | 102 const HloModuleConfig& hlo_module_config);
|
| elemental_ir_emitter.h | 29 #include "tensorflow/compiler/xla/service/hlo_module_config.h"
45 GpuElementalIrEmitter(const HloModuleConfig& hlo_module_config,
|
| elemental_ir_emitter.cc | 71 const HloModuleConfig& hlo_module_config, llvm::Module* module,
73 : ElementalIrEmitter(hlo_module_config, module, b),
74 hlo_module_config_(hlo_module_config),
|
| ir_emitter.h | 113 explicit IrEmitter(const HloModuleConfig& hlo_module_config,
|
| ir_emitter_unnested.h | 146 IrEmitterUnnested(const HloModuleConfig& hlo_module_config,
|
| ir_emitter.cc | 57 IrEmitter::IrEmitter(const HloModuleConfig& hlo_module_config,
65 hlo_module_config_(hlo_module_config) {
[all...] |
| ir_emitter_unnested.cc | 174 IrEmitterUnnested::IrEmitterUnnested(const HloModuleConfig& hlo_module_config,
177 : IrEmitter(hlo_module_config, ir_emitter_context, /*is_nested=*/false),
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| elemental_ir_emitter.h | 25 #include "tensorflow/compiler/xla/service/hlo_module_config.h"
37 ElementalIrEmitter(const HloModuleConfig& hlo_module_config,
39 : b_(b), module_(module), hlo_module_config_(hlo_module_config) {}
|
