| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | elemental_ir_emitter.h | 32 CpuElementalIrEmitter(const HloModuleConfig& module_config, in CpuElementalIrEmitter()
|
| D | ir_function.cc | 47 const HloModuleConfig& module_config, in IrFunction()
73 const HloModuleConfig& module_config) { in Initialize()
|
| D | tiled_dot_emitter.cc | 1006 const HloModuleConfig& module_config) { in EmitRowMajorGemv()
1034 const HloModuleConfig& module_config) { in EmitColumnMajorGemv()
1062 const HloModuleConfig& module_config) { in EmitSmallGemm()
|
| D | cpu_compiler.cc | 717 const HloModuleConfig& module_config) { in CompilerTargetOptions()
724 llvm::CodeGenOpt::Level CodeGenOptLevel(const HloModuleConfig& module_config) { in CodeGenOptLevel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_proto_util.cc | 43 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateModuleFromProto()
|
| D | hlo_module.cc | 390 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateFromProto()
527 HloModuleConfig module_config(ProgramShape{program_shape}); in CreateModuleConfigFromShape() local
|
| D | executable.h | 355 const HloModuleConfig& module_config() const { return hlo_module_->config(); } in module_config() function
|
| D | instruction_fusion.cc | 502 HloModuleConfig module_config; in Run() local
|
| D | service.cc | 806 std::unique_ptr<HloModuleConfig> module_config, Backend* backend, in BuildExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 195 bool MaybeLoadPtxFromFile(const HloModuleConfig module_config, in MaybeLoadPtxFromFile()
328 NVPTXCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()
|
| D | amdgpu_compiler.cc | 134 AMDGPUCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()
|
| D | gpu_compiler.cc | 1098 GpuCompiler::CompileToTargetBinary(const HloModuleConfig& module_config, in CompileToTargetBinary()
1611 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 | 78 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
|
| D | llvm_util.cc | 507 llvm::FastMathFlags GetCpuFastMathFlags(const HloModuleConfig& module_config) { in GetCpuFastMathFlags()
612 const HloModuleConfig& module_config, in CreateCpuFunction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 66 HloModuleConfig module_config; in CompileMlirModule() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 63 const HloModuleConfig& module_config, llvm::Module* llvm_module, in CompileTargetBinary()
|
| D | test_utils.cc | 745 const HloModuleConfig& module_config = module->config(); in MakeFakeArguments() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tf_xla_mlir_translate.cc | 84 const xla::HloModuleConfig module_config( in PrintHloModuleText() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | c_api_decl.h | 328 XLA_HloModuleConfig* module_config; member
333 XLA_HloModuleConfig module_config; member
|