Home
last modified time | relevance | path

Searched defs:module_config (Results 1 – 21 of 21) 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.cc472 const HloModuleConfig& module_config) { in CompilerTargetOptions()
479 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()
Dhlo_module.cc331 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateFromProto()
438 HloModuleConfig module_config(ProgramShape{program_shape}); in CreateModuleConfigFromShape() local
Dexecutable.h357 const HloModuleConfig& module_config() const { return hlo_module_->config(); } in module_config() function
Dinstruction_fusion.cc481 HloModuleConfig module_config; in Run() local
Dservice.cc758 std::unique_ptr<HloModuleConfig> module_config, Backend* backend, in BuildExecutable()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc203 bool MaybeLoadPtxFromFile(const HloModuleConfig module_config, in MaybeLoadPtxFromFile()
301 NVPTXCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()
Damdgpu_compiler.cc114 AMDGPUCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()
Dgpu_compiler.cc676 GpuCompiler::CompileToTargetBinary(const HloModuleConfig& module_config, in CompileToTargetBinary()
1098 const HloModuleConfig& module_config, in CompileLmhloToExecutable()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.h277 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
290 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
Dkernel_support_library.cc77 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
Dllvm_util.cc506 llvm::FastMathFlags GetCpuFastMathFlags(const HloModuleConfig& module_config) { in GetCpuFastMathFlags()
612 const HloModuleConfig& module_config, in CreateCpuFunction()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc73 HloModuleConfig module_config; in CompileMlirModule() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_compiler_test.cc61 const HloModuleConfig& module_config, llvm::Module* llvm_module, in CompileTargetBinary()
Dtest_utils.cc688 const HloModuleConfig& module_config = module->config(); in MakeFakeArguments() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtf_xla_mlir_translate.cc70 const xla::HloModuleConfig module_config( in PrintHloModuleText() local
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h298 XLA_HloModuleConfig* module_config; member
303 XLA_HloModuleConfig module_config; member
/external/llvm-project/lldb/source/Plugins/ExpressionParser/Clang/
DClangUserExpression.cpp554 CppModuleConfiguration module_config = GetModuleConfig(m_language, exe_ctx); in PrepareForParsing() local