Home
last modified time | relevance | path

Searched defs:module_config (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Delemental_ir_emitter.h32 CpuElementalIrEmitter(const HloModuleConfig& module_config, in CpuElementalIrEmitter()
Dir_function.cc46 const HloModuleConfig& module_config, in IrFunction()
72 const HloModuleConfig& module_config) { in Initialize()
Dcpu_compiler.cc410 const HloModuleConfig& module_config) { in CompilerTargetOptions()
429 llvm::CodeGenOpt::Level CodeGenOptLevel(const HloModuleConfig& module_config) { in CodeGenOptLevel()
Dtiled_dot_emitter.cc994 const HloModuleConfig& module_config) { in EmitRowMajorGemv()
1021 const HloModuleConfig& module_config) { in EmitColumnMajorGemv()
1049 const HloModuleConfig& module_config) { in EmitSmallGemm()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_proto_util.cc41 const HloModuleProto& proto, const HloModuleConfig& module_config) { in CreateModuleFromProto()
Dexecutable.h169 const HloModuleConfig& module_config() const { return hlo_module_->config(); } in module_config() function
Dhlo_module.cc274 const HloModuleProto& proto, const HloModuleConfig& module_config) { in CreateFromProto()
375 HloModuleConfig module_config(ProgramShape{program_shape}); in CreateModuleConfigFromProto() local
Dservice.cc785 std::unique_ptr<HloModuleConfig> module_config, Backend* backend, in BuildExecutable()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.h272 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
285 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
Dkernel_support_library.cc73 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
Dllvm_util.cc503 llvm::FastMathFlags GetCpuFastMathFlags(const HloModuleConfig& module_config) { in GetCpuFastMathFlags()
610 const HloModuleConfig& module_config, in CreateCpuFunction()