| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group.cc | 20 HloModuleGroup::HloModuleGroup(std::unique_ptr<HloModule> module) in HloModuleGroup() function in xla::HloModuleGroup
25 HloModuleGroup::HloModuleGroup(absl::string_view name, in HloModuleGroup() function in xla::HloModuleGroup
33 HloModuleGroup::HloModuleGroup( in HloModuleGroup() function in xla::HloModuleGroup
41 std::vector<std::unique_ptr<HloModule>> HloModuleGroup::ConsumeModules() { in ConsumeModules()
50 string HloModuleGroup::ToString() const { in ToString()
59 HloModuleGroupProto HloModuleGroup::ToProto() const { in ToProto()
68 /* static */ StatusOr<HloModuleGroup> HloModuleGroup::CreateFromProto( in CreateFromProto()
85 return HloModuleGroup(proto.name(), absl::MakeSpan(modules)); in CreateFromProto()
88 void HloModuleGroup::push_back(std::unique_ptr<HloModule> module) { in push_back()
93 void HloModuleGroup::ReplaceModule(int index, in ReplaceModule()
[all …]
|
| D | hlo_module_group.h | 32 class HloModuleGroup {
35 explicit HloModuleGroup(absl::string_view name) : name_(name) {} in HloModuleGroup() function
38 explicit HloModuleGroup(std::unique_ptr<HloModule> module);
41 HloModuleGroup(absl::string_view name,
43 HloModuleGroup(absl::string_view name,
69 static StatusOr<HloModuleGroup> CreateFromProto(
90 std::ostream& operator<<(std::ostream& out, const HloModuleGroup& group);
|
| D | hlo_module_group_test.cc | 49 HloModuleGroup group(std::move(module)); in TEST_F()
56 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup group_copy, in TEST_F()
57 HloModuleGroup::CreateFromProto( in TEST_F()
93 HloModuleGroup group(TestName(), absl::MakeSpan(modules)); in TEST_F()
101 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup group_copy, in TEST_F()
102 HloModuleGroup::CreateFromProto( in TEST_F()
129 HloModuleGroup group(TestName()); in TEST_F()
175 HloModuleGroup group(TestName()); in TEST_F()
|
| 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 | llvm_compiler.h | 73 HloModuleGroup* module_group,
78 std::unique_ptr<HloModuleGroup> module_group,
83 std::unique_ptr<HloModuleGroup> module_group,
|
| D | llvm_compiler.cc | 25 HloModuleGroup* module_group, in RunHloPassesOnModuleGroup()
34 std::unique_ptr<HloModuleGroup> module_group, in RunBackendOnModuleGroup()
42 std::unique_ptr<HloModuleGroup> module_group, in Compile()
|
| 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()
|
| D | hlo_pass_pipeline_test.cc | 31 StatusOr<HloModuleGroup> ParseModuleGroup( in ParseModuleGroup()
33 HloModuleGroup group(TestName()); in ParseModuleGroup()
65 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
162 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup module_group, in TEST_F()
|
| D | hlo_pass_interface.h | 43 virtual StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) = 0;
51 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
|
| D | hlo_pass_pipeline.cc | 107 void HloPassPipeline::MaybeDumpHlo(const HloModuleGroup& module_group, in MaybeDumpHlo()
125 StatusOr<bool> HloPassPipeline::RunOnModuleGroup(HloModuleGroup* module_group) { in RunOnModuleGroup()
|
| D | hlo_pass_fix.h | 59 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
|
| D | compile_only_service.cc | 96 absl::make_unique<HloModuleGroup>(hlo_modules[0]->name(), in CompileAheadOfTime()
|
| D | compiler.cc | 48 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| D | hlo_runner.cc | 429 auto module_group = absl::make_unique<HloModuleGroup>(std::move(module)); in CreateExecutable()
|
| D | service.cc | 368 absl::make_unique<HloModuleGroup>(module_protos[0]->name()); in BuildExecutables()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| 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,
|
| D | compiler.cc | 106 HloModuleGroup* module_group, in RunHloPassesOnModuleGroup()
138 std::unique_ptr<HloModuleGroup> module_group, in RunBackendOnModuleGroup()
146 std::unique_ptr<HloModuleGroup> module_group, in Compile()
174 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | codegen_test_base.cc | 35 auto module_group = absl::make_unique<HloModuleGroup>(std::move(hlo_module)); in CompileToAotCompilationResult()
|
| D | llvm_compiler_test.cc | 96 auto module_group = absl::make_unique<HloModuleGroup>("test_module_group"); in TestMultiModuleCompilation()
|
| /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()
|
| /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()
|