| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.h | 34 const HloModuleConfig& module_config);
41 const HloModuleConfig& module_config);
50 llvm::IRBuilder<>* b, const HloModuleConfig& module_config);
|
| D | elemental_ir_emitter.h | 32 CpuElementalIrEmitter(const HloModuleConfig& module_config, in CpuElementalIrEmitter() argument
34 : ElementalIrEmitter(module_config, module, ir_emitter->b()), in CpuElementalIrEmitter()
|
| D | ir_function.cc | 46 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()
|
| D | ir_function.h | 56 const HloModuleConfig& module_config, llvm::Module* llvm_module,
95 const HloModuleConfig& module_config);
|
| D | cpu_compiler.cc | 410 const HloModuleConfig& module_config) { in CompilerTargetOptions() argument
413 if (module_config.debug_options().xla_cpu_enable_fast_math()) { in CompilerTargetOptions()
416 module_config.debug_options().xla_cpu_fast_math_honor_infs(); in CompilerTargetOptions()
418 module_config.debug_options().xla_cpu_fast_math_honor_nans(); in CompilerTargetOptions()
429 llvm::CodeGenOpt::Level CodeGenOptLevel(const HloModuleConfig& module_config) { in CodeGenOptLevel() argument
431 << module_config.debug_options().xla_backend_optimization_level(); in CodeGenOptLevel()
432 switch (module_config.debug_options().xla_backend_optimization_level()) { in CodeGenOptLevel()
|
| D | tiled_dot_emitter.cc | 994 const HloModuleConfig& module_config) { in EmitRowMajorGemv() argument
1004 module_config, b, config.GetCacheKey(), in EmitRowMajorGemv()
1021 const HloModuleConfig& module_config) { in EmitColumnMajorGemv() argument
1031 module_config, b, config.GetCacheKey(), in EmitColumnMajorGemv()
1049 const HloModuleConfig& module_config) { in EmitSmallGemm() argument
1059 module_config, b, config.GetCacheKey(), lhs, rhs, result, in EmitSmallGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module.cc | 274 const HloModuleProto& proto, const HloModuleConfig& module_config) { in CreateFromProto() argument
284 module_config.entry_computation_layout().parameter_count()); in CreateFromProto()
287 module_config.entry_computation_layout().parameter_layout(i).shape(); in CreateFromProto()
297 module_config.entry_computation_layout().result_layout().shape(); in CreateFromProto()
326 auto module = absl::make_unique<HloModule>(proto.name(), module_config); in CreateFromProto()
375 HloModuleConfig module_config(ProgramShape{program_shape}); in CreateModuleConfigFromProto() local
376 module_config.set_debug_options(debug_options); in CreateModuleConfigFromProto()
381 module_config.mutable_entry_computation_layout(); in CreateModuleConfigFromProto()
389 return module_config; in CreateModuleConfigFromProto()
|
| D | hlo_proto_util.cc | 41 const HloModuleProto& proto, const HloModuleConfig& module_config) { in CreateModuleFromProto() argument
44 HloModule::CreateFromProto(proto, module_config)); in CreateModuleFromProto()
|
| D | local_service.cc | 168 std::unique_ptr<HloModuleConfig> module_config, in CompileExecutable()
172 << module_config->entry_computation_layout().ToString(); in CompileExecutable()
178 return BuildExecutable(proto, std::move(module_config), in CompileExecutable()
|
| D | service.cc | 442 executables[i]->module_config().debug_options().xla_hlo_profile() && in ExecuteParallelAndRegisterResult()
660 std::unique_ptr<HloModuleConfig> module_config, in ExecuteGraphParallel()
666 << module_config->entry_computation_layout().ToString(); in ExecuteGraphParallel()
672 module_configs.push_back(std::move(module_config)); in ExecuteGraphParallel()
785 std::unique_ptr<HloModuleConfig> module_config, Backend* backend, in BuildExecutable() argument
792 CreateModuleFromProto(module_proto, *module_config)); in BuildExecutable()
803 const auto& debug_opts = module_config->debug_options(); in BuildExecutable()
836 std::unique_ptr<HloModuleConfig> module_config, in Compile()
840 << module_config->entry_computation_layout().ToString(); in Compile()
844 BuildExecutable(arg->computation(), std::move(module_config), in Compile()
[all …]
|
| D | compile_only_service.cc | 83 std::unique_ptr<HloModuleConfig> module_config, in CompileAheadOfTime()
90 HloModule::CreateFromProto(instance.computation, *module_config)); in CompileAheadOfTime()
|
| D | hlo_proto_util.h | 42 const HloModuleProto& proto, const HloModuleConfig& module_config);
|
| D | executable.cc | 77 module_config().debug_options().xla_hlo_profile() && in ExecuteOnStreamWrapper()
|
| D | executable.h | 169 const HloModuleConfig& module_config() const { return hlo_module_->config(); } in module_config() function
|
| D | hlo_module.h | 208 const HloModuleProto& proto, const HloModuleConfig& module_config);
|
| D | service.h | 235 std::unique_ptr<HloModuleConfig> module_config, Backend* backend,
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_support_library.h | 266 const HloModuleConfig& module_config, llvm::IRBuilder<>* b,
272 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel() argument
277 EmitAndCallOutlinedKernel(module_config, b, kernel_name, {arg0, arg1, arg2}, in EmitAndCallOutlinedKernel()
285 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel() argument
291 module_config, b, kernel_name, {arg0, arg1, arg2, arg3}, in EmitAndCallOutlinedKernel()
|
| D | llvm_util.cc | 503 llvm::FastMathFlags GetCpuFastMathFlags(const HloModuleConfig& module_config) { in GetCpuFastMathFlags() argument
505 if (!module_config.debug_options().xla_cpu_enable_fast_math()) { in GetCpuFastMathFlags()
513 if (module_config.debug_options().xla_cpu_fast_math_honor_nans()) { in GetCpuFastMathFlags()
517 if (module_config.debug_options().xla_cpu_fast_math_honor_infs()) { in GetCpuFastMathFlags()
610 const HloModuleConfig& module_config, in CreateCpuFunction() argument
622 if (module_config.debug_options().xla_cpu_enable_fast_math()) { in CreateCpuFunction()
626 if (!module_config.debug_options().xla_cpu_fast_math_honor_nans()) { in CreateCpuFunction()
630 if (!module_config.debug_options().xla_cpu_fast_math_honor_infs()) { in CreateCpuFunction()
638 if (cpu::options::OptimizeForSizeRequested(module_config)) { in CreateCpuFunction()
|
| D | kernel_support_library.cc | 73 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel() argument
106 module_config, kernel_name, module); in EmitAndCallOutlinedKernel()
|
| D | llvm_util.h | 266 llvm::FastMathFlags GetCpuFastMathFlags(const HloModuleConfig& module_config);
287 const HloModuleConfig& module_config,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 376 TF_ASSIGN_OR_RETURN(const HloModuleConfig module_config, in GetHloText()
381 HloModule::CreateFromProto(computation_.proto(), module_config)); in GetHloText()
389 TF_ASSIGN_OR_RETURN(const HloModuleConfig module_config, in GetHloDotGraph()
394 HloModule::CreateFromProto(computation_.proto(), module_config)); in GetHloDotGraph()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 58 executable_->module_config().entry_computation_layout(); in ValidateExecutionOptions()
|