Searched defs:module_config (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util_test.cc | 67 const xla::HloModuleConfig module_config( in TEST() local
139 const xla::HloModuleConfig module_config( in TEST() local
177 const xla::HloModuleConfig module_config( in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | elemental_ir_emitter.h | 32 CpuElementalIrEmitter(const HloModuleConfig& module_config, in CpuElementalIrEmitter()
|
| D | ir_function.cc | 46 const HloModuleConfig& module_config, in IrFunction()
72 const HloModuleConfig& module_config) { in Initialize()
|
| D | cpu_compiler.cc | 407 const HloModuleConfig& module_config) { in CompilerTargetOptions()
414 llvm::CodeGenOpt::Level CodeGenOptLevel(const HloModuleConfig& module_config) { in CodeGenOptLevel()
|
| D | tiled_dot_emitter.cc | 994 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/ |
| D | hlo_proto_util.cc | 41 const HloModuleProto& proto, const HloModuleConfig& module_config) { in CreateModuleFromProto()
|
| D | executable.h | 235 const HloModuleConfig& module_config() const { return hlo_module_->config(); } in module_config() function
|
| D | hlo_module.cc | 298 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateFromProto()
398 HloModuleConfig module_config(ProgramShape{program_shape}); in CreateModuleConfigFromShape() local
|
| D | instruction_fusion.cc | 473 HloModuleConfig module_config; in Run() local
|
| D | service.cc | 805 std::unique_ptr<HloModuleConfig> module_config, Backend* backend, in BuildExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_support_library.h | 272 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
285 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
|
| D | kernel_support_library.cc | 73 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
|
| D | llvm_util.cc | 496 llvm::FastMathFlags GetCpuFastMathFlags(const HloModuleConfig& module_config) { in GetCpuFastMathFlags()
598 const HloModuleConfig& module_config, in CreateCpuFunction()
|