| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_pass_pipeline.cc | 46 void RecordPassStartMetadata(HloModuleGroup& module_group, in RecordPassStartMetadata()
76 Status AttemptRecordPassEndMetadata(HloModuleGroup& module_group, in AttemptRecordPassEndMetadata()
91 void RecordPassEndMetadata(HloModuleGroup& module_group, in RecordPassEndMetadata()
117 void SetInstructionMetadata(HloModuleGroup& module_group) { in SetInstructionMetadata()
263 HloModuleGroup& module_group, absl::string_view after_pass_name, in MaybeDumpHloAndSaveFilenames()
283 HloModuleGroup* module_group, in RunOnModuleGroup()
|
| H A D | hlo_pass_interface.h | 135 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) { in RunOnModuleGroup()
150 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group, in RunOnModuleGroup()
|
| H A D | hlo_pass_pipeline_test.cc | 96 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group, in RunOnModuleGroup()
368 HloModuleGroup module_group(TestName()); in TEST_F() local
|
| H A D | llvm_compiler.cc | 26 std::unique_ptr<HloModuleGroup> module_group, in Compile()
|
| H A D | hlo_pass_fix.h | 62 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group, in RunOnModuleGroup()
|
| H A D | compiler.cc | 46 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| H A D | hlo_pass_pipeline.h | 145 HloPassInterface* pass, HloModuleGroup* module_group, in RunHelper()
|
| H A D | service.cc | 303 auto module_group = in BuildExecutables() local
349 auto module_group = in BuildAotResults() local
|
| H A D | compiler.h | 298 std::unique_ptr<HloModuleGroup> module_group, in Compile()
|
| H A D | hlo_runner.cc | 459 auto module_group = std::make_unique<HloModuleGroup>(std::move(module)); in CreateExecutable() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | compiler.cc | 134 std::unique_ptr<HloModuleGroup> module_group, in Compile()
160 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_on_demand_compiler.cc | 112 std::unique_ptr<HloModuleGroup> module_group, in Compile()
187 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | codegen_test_base.cc | 35 auto module_group = std::make_unique<HloModuleGroup>(std::move(hlo_module)); in CompileToAotCompilationResult() local
|
| H A D | llvm_compiler_test.cc | 140 auto module_group = std::make_unique<HloModuleGroup>("test_module_group"); in TestMultiModuleCompilation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_aot_compilation_test.cc | 45 auto module_group = std::make_unique<HloModuleGroup>(std::move(module)); in TEST_F() local
|
| H A D | gpu_compiler.cc | 1436 GpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | vectorized_reduce_with_no_vector_registers_test.cc | 74 auto module_group = std::make_unique<HloModuleGroup>("group"); in TEST_F() local
|
| H A D | cpu_compiler.cc | 308 std::unique_ptr<HloModuleGroup> module_group, in Compile()
1318 CpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|