| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_pass_pipeline.cc | 107 void HloPassPipeline::MaybeDumpHlo(const HloModuleGroup& module_group, in MaybeDumpHlo() argument
110 for (const HloModule* module : module_group.modules()) { in MaybeDumpHlo()
125 StatusOr<bool> HloPassPipeline::RunOnModuleGroup(HloModuleGroup* module_group) { in RunOnModuleGroup() argument
129 << module_group->name() << ": " << name(); in RunOnModuleGroup()
131 if (module_group->modules().empty()) { in RunOnModuleGroup()
137 module_group, in RunOnModuleGroup()
138 GetEnabledPasses(module_group->module(0).config().debug_options())); in RunOnModuleGroup()
|
| D | llvm_compiler.cc | 25 HloModuleGroup* module_group, in RunHloPassesOnModuleGroup() argument
34 std::unique_ptr<HloModuleGroup> module_group, in RunBackendOnModuleGroup() argument
42 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
59 module_group->ConsumeModules(); in Compile()
|
| D | hlo_pass_pipeline_test.cc | 65 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup() argument
67 for (HloModule* module : module_group->modules()) { in RunOnModuleGroup()
162 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup module_group, in TEST_F()
170 module_group.module(0).entry_computation()->root_instruction(); in TEST_F()
172 module_group.module(1).entry_computation()->root_instruction(); in TEST_F()
177 pipeline.RunOnModuleGroup(&module_group)); in TEST_F()
|
| D | hlo_pass_pipeline.h | 66 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override;
76 void MaybeDumpHlo(const HloModuleGroup& module_group,
100 HloModuleGroup* module_group) { in RunHelper() argument
101 return pass->RunOnModuleGroup(module_group); in RunHelper()
|
| D | compiler.h | 155 HloModuleGroup* module_group,
177 std::unique_ptr<HloModuleGroup> module_group,
190 std::unique_ptr<HloModuleGroup> module_group,
217 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
223 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | hlo_pass_interface.h | 43 virtual StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) = 0;
51 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup() argument
53 for (HloModule* module : module_group->modules()) { in RunOnModuleGroup()
|
| D | hlo_pass_fix.h | 59 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup() argument
64 for (const HloModule* module : module_group->modules()) { in RunOnModuleGroup()
70 Pass::RunOnModuleGroup(module_group)); in RunOnModuleGroup()
|
| D | llvm_compiler.h | 73 HloModuleGroup* module_group,
78 std::unique_ptr<HloModuleGroup> module_group,
83 std::unique_ptr<HloModuleGroup> module_group,
|
| D | compiler.cc | 48 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
56 return CompileAheadOfTime(std::move(module_group), options); in CompileAheadOfTime()
|
| D | hlo_runner.cc | 429 auto module_group = absl::make_unique<HloModuleGroup>(std::move(module)); in CreateExecutable() local
432 backend().compiler()->Compile(std::move(module_group), in CreateExecutable()
|
| D | service.cc | 367 auto module_group = in BuildExecutables() local
374 module_group->push_back(std::move(module)); in BuildExecutables()
379 backend->compiler()->Compile(std::move(module_group), in BuildExecutables()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 106 HloModuleGroup* module_group, in RunHloPassesOnModuleGroup() argument
138 std::unique_ptr<HloModuleGroup> module_group, in RunBackendOnModuleGroup() argument
146 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
149 if (module_group->empty()) { in Compile()
152 if (module_group->size() > 1) { in Compile()
160 auto hlo_modules = module_group->ConsumeModules(); in Compile()
174 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
|
| D | compiler.h | 50 HloModuleGroup* module_group,
58 std::unique_ptr<HloModuleGroup> module_group,
63 std::unique_ptr<HloModuleGroup> module_group,
68 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 96 auto module_group = absl::make_unique<HloModuleGroup>("test_module_group"); in TestMultiModuleCompilation() local
97 module_group->push_back(hlo_module->Clone()); in TestMultiModuleCompilation()
98 module_group->push_back(std::move(hlo_module)); in TestMultiModuleCompilation()
104 EXPECT_IS_OK(compiler->Compile(std::move(module_group), in TestMultiModuleCompilation()
|
| D | codegen_test_base.cc | 35 auto module_group = absl::make_unique<HloModuleGroup>(std::move(hlo_module)); in CompileToAotCompilationResult() local
38 backend().compiler()->CompileAheadOfTime(std::move(module_group), in CompileToAotCompilationResult()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.h | 62 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | nvptx_compiler.cc | 880 NVPTXCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.h | 145 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | cpu_compiler.cc | 729 CpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
731 TF_RET_CHECK(!module_group->empty()); in CompileAheadOfTime()
733 module_group->ConsumeModules(); in CompileAheadOfTime()
|