Home
last modified time | relevance | path

Searched refs:module_config (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module.cc331 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateFromProto() argument
342 module_config.entry_computation_layout().parameter_count()); in CreateFromProto()
345 module_config.entry_computation_layout().parameter_layout(i).shape(); in CreateFromProto()
355 module_config.entry_computation_layout().result_layout().shape(); in CreateFromProto()
385 auto module = absl::make_unique<HloModule>(proto.name(), module_config); in CreateFromProto()
438 HloModuleConfig module_config(ProgramShape{program_shape}); in CreateModuleConfigFromShape() local
439 module_config.set_debug_options(debug_options); in CreateModuleConfigFromShape()
442 module_config.set_replica_count(execution_options->num_replicas()); in CreateModuleConfigFromShape()
445 module_config.set_num_partitions(execution_options->num_partitions()); in CreateModuleConfigFromShape()
447 module_config.set_use_spmd_partitioning( in CreateModuleConfigFromShape()
[all …]
Dlocal_service.cc150 std::unique_ptr<HloModuleConfig> module_config, in CompileExecutables()
154 << module_config->entry_computation_layout().ToString(); in CompileExecutables()
165 BuildExecutable(proto, std::move(module_config), in CompileExecutables()
175 module_configs.push_back(std::move(module_config)); in CompileExecutables()
Dhlo_proto_util.cc41 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateModuleFromProto() argument
45 HloModule::CreateFromProto(proto, module_config)); in CreateModuleFromProto()
Dservice.cc392 executables[i]->module_config().debug_options().xla_hlo_profile() && in ExecuteParallelAndRegisterResult()
612 std::unique_ptr<HloModuleConfig> module_config, in ExecuteGraphParallel()
618 << module_config->entry_computation_layout().ToString(); in ExecuteGraphParallel()
624 module_configs.push_back(std::move(module_config)); in ExecuteGraphParallel()
758 std::unique_ptr<HloModuleConfig> module_config, Backend* backend, in BuildExecutable() argument
767 CreateModuleFromProto(module_proto, *module_config, run_backend_only)); in BuildExecutable()
779 const auto& debug_opts = module_config->debug_options(); in BuildExecutable()
812 std::unique_ptr<HloModuleConfig> module_config, in Compile()
816 << module_config->entry_computation_layout().ToString(); in Compile()
820 BuildExecutable(arg->computation(), std::move(module_config), in Compile()
[all …]
Dcompile_only_service.cc104 std::unique_ptr<HloModuleConfig> module_config, in CompileAheadOfTime()
111 HloModule::CreateFromProto(instance.computation, *module_config)); in CompileAheadOfTime()
Dhlo_runner_interface.cc82 HloModuleConfig module_config, in ReadModuleFromModuleBinaryProtofile()
85 return HloModule::CreateFromProto(module_proto, module_config); in ReadModuleFromModuleBinaryProtofile()
Dhlo_proto_util.h45 const HloModuleProto& proto, const HloModuleConfig& module_config,
Dinstruction_fusion.cc481 HloModuleConfig module_config; in Run() local
483 module_config = module->config(); in Run()
484 fusion_config = module_config.mutable_fusion_config(); in Run()
610 module->set_config(module_config); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc203 bool MaybeLoadPtxFromFile(const HloModuleConfig module_config, in MaybeLoadPtxFromFile() argument
210 module_config.debug_options().xla_gpu_ptx_file()) { in MaybeLoadPtxFromFile()
220 if (!module_config.debug_options().xla_gpu_ptx_file().empty() && in MaybeLoadPtxFromFile()
301 NVPTXCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary() argument
318 cached_libdevice_dir_ = GetLibdeviceDir(module_config); in CompileTargetBinary()
326 MaybeLoadPtxFromFile(module_config, debug_module, &ptx))) { in CompileTargetBinary()
330 module_config, libdevice_dir)); in CompileTargetBinary()
335 module_config, relocatable); in CompileTargetBinary()
Damdgpu_compiler.cc114 AMDGPUCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary() argument
122 rocdl_dir_ = GetROCDLDir(module_config); in CompileTargetBinary()
134 hsaco, amdgpu::CompileToHsaco(llvm_module, gpu_version, module_config, in CompileTargetBinary()
Dgpu_compiler.h89 CompileTargetBinary(const HloModuleConfig& module_config,
105 const HloModuleConfig& module_config,
169 const HloModuleConfig& module_config,
Dgpu_compiler.cc676 GpuCompiler::CompileToTargetBinary(const HloModuleConfig& module_config, in CompileToTargetBinary() argument
684 [this, stream_exec, &module_config, debug_module]( in CompileToTargetBinary()
710 CompileTargetBinary(module_config, llvm_module, gpu_version, in CompileToTargetBinary()
719 module_config.debug_options()); in CompileToTargetBinary()
767 if (module_config.debug_options().xla_gpu_force_compilation_parallelism() != in CompileToTargetBinary()
771 module_config.debug_options().xla_gpu_force_compilation_parallelism()); in CompileToTargetBinary()
1098 const HloModuleConfig& module_config, in CompileLmhloToExecutable() argument
1119 IrEmitterUnnested::Create(module_config, /*hlo_computation=*/nullptr, in CompileLmhloToExecutable()
1140 module_config, std::move(llvm_module), stream_exec, in CompileLmhloToExecutable()
Damdgpu_compiler.h43 const HloModuleConfig& module_config, llvm::Module* llvm_module,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.h34 const HloModuleConfig& module_config);
41 const HloModuleConfig& module_config);
50 llvm::IRBuilder<>* b, const HloModuleConfig& module_config);
Delemental_ir_emitter.h32 CpuElementalIrEmitter(const HloModuleConfig& module_config, in CpuElementalIrEmitter() argument
35 hlo_module_config_(module_config), in CpuElementalIrEmitter()
Dir_function.cc46 const HloModuleConfig& module_config, in IrFunction() argument
53 Initialize(function_name, linkage, module_config); in IrFunction()
72 const HloModuleConfig& module_config) { in Initialize() argument
143 function_ = llvm_ir::CreateCpuFunction(function_type, linkage, module_config, in Initialize()
Dir_function.h56 const HloModuleConfig& module_config, llvm::Module* llvm_module,
95 const HloModuleConfig& module_config);
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc225 ApiConverter::Free(&hlo_module.module_config); in RunHloPasses()
227 hlo_module.module_config = ApiConverter::ToC(module->config()); in RunHloPasses()
253 ApiConverter::Free(&hlo_module.module_config); in RunBackend()
256 hlo_module.module_config = ApiConverter::ToC(module->config()); in RunBackend()
282 se_module_group.module_config = in Compile()
288 ApiConverter::Free(&se_module_group.module_config[i]); in Compile()
290 delete[] se_module_group.module_config; in Compile()
294 se_module_group.module_config[i] = ApiConverter::ToC(config); in Compile()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_module_importer.cc59 auto module_config, in Import()
62 module_proto, module_config)); in Import()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.h271 const HloModuleConfig& module_config, llvm::IRBuilder<>* b,
277 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel() argument
282 EmitAndCallOutlinedKernel(module_config, b, kernel_name, {arg0, arg1, arg2}, in EmitAndCallOutlinedKernel()
290 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel() argument
296 module_config, b, kernel_name, {arg0, arg1, arg2, arg3}, in EmitAndCallOutlinedKernel()
Dkernel_support_library.cc77 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel() argument
110 module_config, kernel_name, module); in EmitAndCallOutlinedKernel()
Dllvm_util.h261 llvm::FastMathFlags GetCpuFastMathFlags(const HloModuleConfig& module_config);
283 const HloModuleConfig& module_config,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc73 HloModuleConfig module_config; in CompileMlirModule() local
74 module_config.set_debug_options(DefaultDebugOptionsIgnoringFlags()); in CompileMlirModule()
77 module_config, Compiler::CompileOptions(), "main", stream_exec, in CompileMlirModule()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h298 XLA_HloModuleConfig* module_config; member
303 XLA_HloModuleConfig module_config; member
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc93 executable->executable()->module_config().entry_computation_layout(); in GetInputTuples()
303 executable->executable()->module_config().entry_computation_layout(); in RunExecutable()
320 ->module_config() in RunExecutable()
323 &executable->executable()->module_config().static_device_assignment()); in RunExecutable()
350 ->module_config() in RunExecutable()

123