| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_function.cc | 47 const HloModuleConfig& module_config, in IrFunction()
73 const HloModuleConfig& module_config) { in Initialize()
|
| H A 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()
|
| H A D | elemental_ir_emitter.h | 32 CpuElementalIrEmitter(const HloModuleConfig& module_config, in CpuElementalIrEmitter()
|
| H A D | cpu_compiler.cc | 717 const HloModuleConfig& module_config) { in CompilerTargetOptions()
724 llvm::CodeGenOpt::Level CodeGenOptLevel(const HloModuleConfig& module_config) { in CodeGenOptLevel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | nvptx_compiler.cc | 195 bool MaybeLoadPtxFromFile(const HloModuleConfig module_config, in MaybeLoadPtxFromFile()
328 NVPTXCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()
|
| H A D | amdgpu_compiler.cc | 134 AMDGPUCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()
|
| H A D | gpu_compiler.cc | 1098 GpuCompiler::CompileToTargetBinary(const HloModuleConfig& module_config, in CompileToTargetBinary()
1611 const HloModuleConfig& module_config, in CompileLmhloToExecutable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A 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()
|
| H A D | llvm_util.cc | 507 llvm::FastMathFlags GetCpuFastMathFlags(const HloModuleConfig& module_config) { in GetCpuFastMathFlags()
612 const HloModuleConfig& module_config, in CreateCpuFunction()
|
| H A D | kernel_support_library.cc | 78 const HloModuleConfig& module_config, llvm::IRBuilder<>* b, in EmitAndCallOutlinedKernel()
|
| /aosp_15_r20/external/pytorch/benchmarks/framework_overhead_benchmark/ |
| H A D | pt_wrapper_module.py | 24 def __init__(self, wrapped_type, module_config, debug, save=False): argument
|
| H A D | framework_overhead_benchmark.py | 38 def benchmark_simple_fn(args, config, module_config, module_type, result): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_proto_util.cc | 43 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateModuleFromProto()
|
| H A D | hlo_module.cc | 390 const HloModuleProto& proto, const HloModuleConfig& module_config, in CreateFromProto()
527 HloModuleConfig module_config(ProgramShape{program_shape}); in CreateModuleConfigFromShape() local
|
| H A D | executable.h | 355 const HloModuleConfig& module_config() const { return hlo_module_->config(); } in module_config() function
|
| H A D | instruction_fusion.cc | 502 HloModuleConfig module_config; in Run() local
|
| H A D | service.cc | 806 std::unique_ptr<HloModuleConfig> module_config, Backend* backend, in BuildExecutable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | mlir_gpu_test_base.cc | 66 HloModuleConfig module_config; in CompileMlirModule() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | llvm_compiler_test.cc | 63 const HloModuleConfig& module_config, llvm::Module* llvm_module, in CompileTargetBinary()
|
| H A D | test_utils.cc | 745 const HloModuleConfig& module_config = module->config(); in MakeFakeArguments() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | c_api_decl.h | 328 XLA_HloModuleConfig* module_config; member
333 XLA_HloModuleConfig module_config; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | tf_xla_mlir_translate.cc | 84 const xla::HloModuleConfig module_config( in PrintHloModuleText() local
|