| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | elemental_ir_emitter.h | 32 CpuElementalIrEmitter(const HloModuleConfig& module_config, in CpuElementalIrEmitter()
|
| D | ir_function.cc | 46 const HloModuleConfig& module_config, in IrFunction()
72 const HloModuleConfig& module_config) { in Initialize()
|
| D | cpu_compiler.cc | 497 const HloModuleConfig& module_config) { in CompilerTargetOptions()
504 llvm::CodeGenOpt::Level CodeGenOptLevel(const HloModuleConfig& module_config) { in CodeGenOptLevel()
|
| D | tiled_dot_emitter.cc | 996 const HloModuleConfig& module_config) { in EmitRowMajorGemv()
1024 const HloModuleConfig& module_config) { in EmitColumnMajorGemv()
1052 const HloModuleConfig& module_config) { in EmitSmallGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_proto_util.cc | 41 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateModuleFromProto()
|
| D | hlo_module.cc | 338 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateFromProto()
447 HloModuleConfig module_config(ProgramShape{program_shape}); in CreateModuleConfigFromShape() local
|
| D | executable.h | 357 const HloModuleConfig& module_config() const { return hlo_module_->config(); } in module_config() function
|
| D | instruction_fusion.cc | 492 HloModuleConfig module_config; in Run() local
|
| D | service.cc | 744 std::unique_ptr<HloModuleConfig> module_config, Backend* backend, in BuildExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 222 bool MaybeLoadPtxFromFile(const HloModuleConfig module_config, in MaybeLoadPtxFromFile()
355 NVPTXCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()
|
| D | amdgpu_compiler.cc | 128 AMDGPUCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()
|
| D | gpu_compiler.cc | 843 GpuCompiler::CompileToTargetBinary(const HloModuleConfig& module_config, in CompileToTargetBinary()
1289 const HloModuleConfig& module_config, in CompileLmhloToExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_support_library.h | 279 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
292 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
|
| D | kernel_support_library.cc | 77 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
|
| D | llvm_util.cc | 506 llvm::FastMathFlags GetCpuFastMathFlags(const HloModuleConfig& module_config) { in GetCpuFastMathFlags()
628 const HloModuleConfig& module_config, in CreateCpuFunction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 63 HloModuleConfig module_config; in CompileMlirModule() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 61 const HloModuleConfig& module_config, llvm::Module* llvm_module, in CompileTargetBinary()
|
| D | test_utils.cc | 690 const HloModuleConfig& module_config = module->config(); in MakeFakeArguments() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tf_xla_mlir_translate.cc | 70 const xla::HloModuleConfig module_config( in PrintHloModuleText() local
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_decl.h | 301 XLA_HloModuleConfig* module_config; member
306 XLA_HloModuleConfig module_config; member
|