Home
last modified time | relevance | path

Searched refs:module_group (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_pass_pipeline.cc44 void RecordPassStartMetadata(HloModuleGroup& module_group, in RecordPassStartMetadata() argument
47 for (HloModule* module : module_group.modules()) { in RecordPassStartMetadata()
74 Status AttemptRecordPassEndMetadata(HloModuleGroup& module_group, in AttemptRecordPassEndMetadata() argument
77 for (HloModule* module : module_group.modules()) { in AttemptRecordPassEndMetadata()
78 for (HloModule* other_module : module_group.modules()) { in AttemptRecordPassEndMetadata()
89 void RecordPassEndMetadata(HloModuleGroup& module_group, in RecordPassEndMetadata() argument
92 AttemptRecordPassEndMetadata(module_group, pass_name, module_changed); in RecordPassEndMetadata()
115 void SetInstructionMetadata(HloModuleGroup& module_group) { in SetInstructionMetadata() argument
116 for (HloModule* module : module_group.modules()) { in SetInstructionMetadata()
250 HloModuleGroup& module_group, absl::string_view after_pass_name, in MaybeDumpHloAndSaveFilenames() argument
[all …]
Dhlo_pass_pipeline_test.cc69 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup() argument
71 for (HloModule* module : module_group->modules()) { in RunOnModuleGroup()
166 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup module_group, in TEST_F()
174 module_group.module(0).entry_computation()->root_instruction(); in TEST_F()
176 module_group.module(1).entry_computation()->root_instruction(); in TEST_F()
181 pipeline.RunOnModuleGroup(&module_group)); in TEST_F()
264 HloModuleGroup module_group(TestName()); in TEST_F() local
265 module_group.push_back(CreateNewVerifiedModule()); in TEST_F()
266 module_group.push_back(CreateNewVerifiedModule()); in TEST_F()
271 TF_ASSERT_OK(pipeline.RunOnModuleGroup(&module_group).status()); in TEST_F()
[all …]
Dhlo_pass_pipeline.h82 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override;
95 void MaybeDumpHloAndSaveFilenames(HloModuleGroup& module_group,
122 HloModuleGroup* module_group) { in RunHelper() argument
123 TF_ASSIGN_OR_RETURN(bool changed, pass->RunOnModuleGroup(module_group)); in RunHelper()
124 module_group->Cleanup(); in RunHelper()
Dhlo_pass_interface.h43 virtual StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) = 0;
53 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup() argument
55 for (HloModule* module : module_group->modules()) { in RunOnModuleGroup()
Dcompiler.h228 std::unique_ptr<HloModuleGroup> module_group,
232 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
235 return Compile(std::move(module_group), stream_exec, in Compile()
262 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
268 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
Dllvm_compiler.cc26 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
43 module_group->ConsumeModules(); in Compile()
Dhlo_pass_fix.h60 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup() argument
68 Pass::RunOnModuleGroup(module_group)); in RunOnModuleGroup()
Dcompiler.cc48 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
56 return CompileAheadOfTime(std::move(module_group), options); in CompileAheadOfTime()
Dllvm_compiler.h74 std::unique_ptr<HloModuleGroup> module_group,
Dhlo_runner.cc429 auto module_group = absl::make_unique<HloModuleGroup>(std::move(module)); in CreateExecutable() local
432 backend().compiler()->Compile(std::move(module_group), in CreateExecutable()
Dservice.cc306 auto module_group = in BuildExecutables() local
314 module_group->push_back(std::move(module)); in BuildExecutables()
320 std::move(module_group), in BuildExecutables()
323 auto modules = module_group->ConsumeModules(); in BuildExecutables()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc129 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
132 if (module_group->empty()) { in Compile()
135 if (module_group->size() > 1) { in Compile()
143 auto hlo_modules = module_group->ConsumeModules(); in Compile()
155 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
Dcompiler.h53 std::unique_ptr<HloModuleGroup> module_group,
58 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc276 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
281 stream_executor::tpu::SerializeProto(module_group->ToProto()); in Compile()
283 new XLA_HloModuleConfig[module_group->size()]; in Compile()
284 int module_group_size = module_group->size(); in Compile()
292 for (int i = 0; i < module_group->size(); ++i) { in Compile()
293 const auto& config = module_group->module(i).config(); in Compile()
312 SE_Executable** se_executables = new SE_Executable*[module_group->size()]; in Compile()
325 for (int i = 0; i < module_group->size(); ++i) { in Compile()
351 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dvectorized_reduce_with_no_vector_registers_test.cc74 auto module_group = absl::make_unique<HloModuleGroup>("group"); in TEST_F() local
75 module_group->push_back(std::move(hlo_module)); in TEST_F()
103 cpu_compiler.CompileAheadOfTime(std::move(module_group), in TEST_F()
Dcpu_compiler.h129 std::unique_ptr<HloModuleGroup> module_group,
148 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
Dcpu_compiler.cc191 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
200 return LLVMCompiler::Compile(std::move(module_group), stream_execs, options); in Compile()
804 CpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
806 TF_RET_CHECK(!module_group->empty()); in CompileAheadOfTime()
808 module_group->ConsumeModules(); in CompileAheadOfTime()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_compiler_test.cc138 auto module_group = absl::make_unique<HloModuleGroup>("test_module_group"); in TestMultiModuleCompilation() local
139 module_group->push_back(hlo_module->Clone()); in TestMultiModuleCompilation()
140 module_group->push_back(std::move(hlo_module)); in TestMultiModuleCompilation()
146 EXPECT_IS_OK(compiler->Compile(std::move(module_group), in TestMultiModuleCompilation()
Dcodegen_test_base.cc35 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/
Dgpu_compiler.h101 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
Dgpu_compiler.cc978 GpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument