| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_pass_pipeline.cc | 46 void RecordPassStartMetadata(HloModuleGroup& module_group, in RecordPassStartMetadata() argument
49 for (HloModule* module : module_group.modules()) { in RecordPassStartMetadata()
76 Status AttemptRecordPassEndMetadata(HloModuleGroup& module_group, in AttemptRecordPassEndMetadata() argument
79 for (HloModule* module : module_group.modules()) { in AttemptRecordPassEndMetadata()
80 for (HloModule* other_module : module_group.modules()) { in AttemptRecordPassEndMetadata()
91 void RecordPassEndMetadata(HloModuleGroup& module_group, in RecordPassEndMetadata() argument
94 AttemptRecordPassEndMetadata(module_group, pass_name, module_changed); in RecordPassEndMetadata()
117 void SetInstructionMetadata(HloModuleGroup& module_group) { in SetInstructionMetadata() argument
118 for (HloModule* module : module_group.modules()) { in SetInstructionMetadata()
263 HloModuleGroup& module_group, absl::string_view after_pass_name, in MaybeDumpHloAndSaveFilenames() argument
[all …]
|
| D | hlo_pass_pipeline_test.cc | 96 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group, in RunOnModuleGroup() argument
100 for (HloModule* module : module_group->modules()) { in RunOnModuleGroup()
270 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup module_group, in TEST_F()
278 module_group.module(0).entry_computation()->root_instruction(); in TEST_F()
280 module_group.module(1).entry_computation()->root_instruction(); in TEST_F()
285 pipeline.RunOnModuleGroup(&module_group)); in TEST_F()
368 HloModuleGroup module_group(TestName()); in TEST_F() local
369 module_group.push_back(CreateNewVerifiedModule()); in TEST_F()
370 module_group.push_back(CreateNewVerifiedModule()); in TEST_F()
375 TF_ASSERT_OK(pipeline.RunOnModuleGroup(&module_group).status()); in TEST_F()
[all …]
|
| D | hlo_pass_interface.h | 135 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) { in RunOnModuleGroup() argument
136 return RunOnModuleGroup(module_group, /*execution_threads=*/{}); in RunOnModuleGroup()
139 HloModuleGroup* module_group,
150 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group, in RunOnModuleGroup() argument
154 for (HloModule* module : module_group->modules()) { in RunOnModuleGroup()
|
| D | hlo_pass_pipeline.h | 87 HloModuleGroup* module_group,
106 void MaybeDumpHloAndSaveFilenames(HloModuleGroup& module_group,
145 HloPassInterface* pass, HloModuleGroup* module_group, in RunHelper() argument
148 bool changed, pass->RunOnModuleGroup(module_group, execution_threads)); in RunHelper()
149 module_group->Cleanup(); in RunHelper()
|
| D | compiler.h | 294 std::unique_ptr<HloModuleGroup> module_group,
298 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
301 return Compile(std::move(module_group), stream_exec, in Compile()
328 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
334 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | llvm_compiler.cc | 26 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
43 module_group->ConsumeModules(); in Compile()
|
| D | compiler.cc | 46 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
54 return CompileAheadOfTime(std::move(module_group), options); in CompileAheadOfTime()
|
| D | hlo_pass_fix.h | 62 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group, in RunOnModuleGroup() argument
72 Pass::RunOnModuleGroup(module_group, execution_threads)); in RunOnModuleGroup()
|
| D | llvm_compiler.h | 74 std::unique_ptr<HloModuleGroup> module_group,
|
| D | service.cc | 303 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()
349 auto module_group = in BuildAotResults() local
357 module_group->push_back(std::move(module)); in BuildAotResults()
367 backend->compiler()->CompileAheadOfTime(std::move(module_group), in BuildAotResults()
|
| D | hlo_runner.cc | 459 auto module_group = std::make_unique<HloModuleGroup>(std::move(module)); in CreateExecutable() local
462 backend().compiler()->Compile(std::move(module_group), in CreateExecutable()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 112 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
117 stream_executor::tpu::SerializeProto(module_group->ToProto()); in Compile()
119 new XLA_HloModuleConfig[module_group->size()]; in Compile()
120 int module_group_size = module_group->size(); in Compile()
128 for (int i = 0; i < module_group->size(); ++i) { in Compile()
129 const auto& config = module_group->module(i).config(); in Compile()
148 SE_Executable** se_executables = new SE_Executable*[module_group->size()]; in Compile()
161 for (int i = 0; i < module_group->size(); ++i) { in Compile()
187 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 134 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
137 if (module_group->empty()) { in Compile()
140 if (module_group->size() > 1) { in Compile()
148 auto hlo_modules = module_group->ConsumeModules(); in Compile()
160 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
|
| D | compiler.h | 52 std::unique_ptr<HloModuleGroup> module_group,
57 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | vectorized_reduce_with_no_vector_registers_test.cc | 74 auto module_group = std::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()
|
| D | cpu_compiler.h | 134 std::unique_ptr<HloModuleGroup> module_group,
150 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | cpu_compiler.cc | 308 std::unique_ptr<HloModuleGroup> module_group, in Compile() argument
317 return LLVMCompiler::Compile(std::move(module_group), stream_execs, options); in Compile()
1318 CpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
1320 TF_RET_CHECK(!module_group->empty()); in CompileAheadOfTime()
1322 module_group->ConsumeModules(); in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 140 auto module_group = std::make_unique<HloModuleGroup>("test_module_group"); in TestMultiModuleCompilation() local
141 module_group->push_back(hlo_module->Clone()); in TestMultiModuleCompilation()
142 module_group->push_back(std::move(hlo_module)); in TestMultiModuleCompilation()
148 EXPECT_IS_OK(compiler->Compile(std::move(module_group), in TestMultiModuleCompilation()
|
| D | codegen_test_base.cc | 35 auto module_group = std::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 | gpu_aot_compilation_test.cc | 45 auto module_group = std::make_unique<HloModuleGroup>(std::move(module)); in TEST_F() local
50 compiler.CompileAheadOfTime(std::move(module_group), aot_options)); in TEST_F()
|
| D | gpu_compiler.h | 100 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | gpu_compiler.cc | 1436 GpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime() argument
1444 module_group->ConsumeModules(); in CompileAheadOfTime()
|