| /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 std::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()
94 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,
75 friend H AbslHashValue(H h, const HloModuleGroup& group) { in AbslHashValue()
84 static StatusOr<HloModuleGroup> CreateFromProto(
112 std::ostream& operator<<(std::ostream& out, const HloModuleGroup& group);
|
| D | hlo_module_group_test.cc | 50 HloModuleGroup group(std::move(module)); in TEST_F()
57 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup group_copy, in TEST_F()
58 HloModuleGroup::CreateFromProto( in TEST_F()
94 HloModuleGroup group(TestName(), absl::MakeSpan(modules)); in TEST_F()
102 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup group_copy, in TEST_F()
103 HloModuleGroup::CreateFromProto( in TEST_F()
130 HloModuleGroup group(TestName()); in TEST_F()
178 HloModuleGroup group(TestName()); in TEST_F()
216 HloModuleGroup group(std::move(old_module)); in TEST_F()
|
| 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()
|
| D | hlo_pass_interface.h | 135 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) { in RunOnModuleGroup()
139 HloModuleGroup* module_group,
150 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group, in RunOnModuleGroup()
|
| D | hlo_pass_pipeline_test.cc | 35 StatusOr<HloModuleGroup> ParseModuleGroup( in ParseModuleGroup()
37 HloModuleGroup group(TestName()); in ParseModuleGroup()
96 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group, in RunOnModuleGroup()
270 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup module_group, in TEST_F()
368 HloModuleGroup module_group(TestName()); in TEST_F()
|
| D | hlo_pass_pipeline.h | 87 HloModuleGroup* module_group,
106 void MaybeDumpHloAndSaveFilenames(HloModuleGroup& module_group,
145 HloPassInterface* pass, HloModuleGroup* module_group, in RunHelper()
|
| D | compiler.h | 294 std::unique_ptr<HloModuleGroup> module_group,
298 std::unique_ptr<HloModuleGroup> module_group, in Compile()
328 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
334 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | llvm_compiler.h | 74 std::unique_ptr<HloModuleGroup> module_group,
|
| D | llvm_compiler.cc | 26 std::unique_ptr<HloModuleGroup> module_group, in Compile()
|
| D | compiler.cc | 46 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| D | hlo_pass_fix.h | 62 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group, in RunOnModuleGroup()
|
| D | compile_only_service.cc | 134 std::make_unique<HloModuleGroup>(hlo_modules[0]->name(), in CompileAheadOfTime()
|
| D | service.cc | 304 std::make_unique<HloModuleGroup>(module_protos[0]->name()); in BuildExecutables()
350 std::make_unique<HloModuleGroup>(module_protos[0]->name()); in BuildAotResults()
|
| D | hlo_runner.cc | 459 auto module_group = std::make_unique<HloModuleGroup>(std::move(module)); in CreateExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.h | 52 std::unique_ptr<HloModuleGroup> module_group,
57 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | compiler.cc | 134 std::unique_ptr<HloModuleGroup> module_group, in Compile()
160 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.h | 134 std::unique_ptr<HloModuleGroup> module_group,
150 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | vectorized_reduce_with_no_vector_registers_test.cc | 74 auto module_group = std::make_unique<HloModuleGroup>("group"); in TEST_F()
|
| D | cpu_compiler.cc | 308 std::unique_ptr<HloModuleGroup> module_group, in Compile()
1318 CpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | codegen_test_base.cc | 35 auto module_group = std::make_unique<HloModuleGroup>(std::move(hlo_module)); in CompileToAotCompilationResult()
|
| D | llvm_compiler_test.cc | 140 auto module_group = std::make_unique<HloModuleGroup>("test_module_group"); in TestMultiModuleCompilation()
|
| /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()
|
| D | gpu_compiler.h | 100 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| /external/tensorflow/tensorflow/core/tpu/ |
| 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()
|