| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group.cc | 22 HloModuleGroup::HloModuleGroup(std::unique_ptr<HloModule> module) in HloModuleGroup() function in xla::HloModuleGroup
27 HloModuleGroup::HloModuleGroup(absl::string_view name, in HloModuleGroup() function in xla::HloModuleGroup
35 HloModuleGroup::HloModuleGroup( in HloModuleGroup() function in xla::HloModuleGroup
43 std::vector<std::unique_ptr<HloModule>> HloModuleGroup::ConsumeModules() { in ConsumeModules()
52 string HloModuleGroup::ToString() const { in ToString()
61 HloModuleGroupProto HloModuleGroup::ToProto() const { in ToProto()
70 uint64 HloModuleGroup::Hash() const { in Hash()
78 /* static */ StatusOr<HloModuleGroup> HloModuleGroup::CreateFromProto( in CreateFromProto()
95 return HloModuleGroup(proto.name(), absl::MakeSpan(modules)); in CreateFromProto()
98 void HloModuleGroup::push_back(std::unique_ptr<HloModule> module) { in push_back()
[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,
78 static StatusOr<HloModuleGroup> CreateFromProto(
99 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()
214 HloModuleGroup group(std::move(old_module)); in TEST_F()
|
| D | hlo_pass_pipeline.cc | 44 void RecordPassStartMetadata(HloModuleGroup& module_group, in RecordPassStartMetadata()
74 Status AttemptRecordPassEndMetadata(HloModuleGroup& module_group, in AttemptRecordPassEndMetadata()
89 void RecordPassEndMetadata(HloModuleGroup& module_group, in RecordPassEndMetadata()
115 void SetInstructionMetadata(HloModuleGroup& module_group) { in SetInstructionMetadata()
250 HloModuleGroup& module_group, absl::string_view after_pass_name, in MaybeDumpHloAndSaveFilenames()
267 StatusOr<bool> HloPassPipeline::RunOnModuleGroup(HloModuleGroup* module_group) { in RunOnModuleGroup()
|
| D | hlo_pass_pipeline_test.cc | 35 StatusOr<HloModuleGroup> ParseModuleGroup( in ParseModuleGroup()
37 HloModuleGroup group(TestName()); in ParseModuleGroup()
69 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
166 TF_ASSERT_OK_AND_ASSIGN(HloModuleGroup module_group, in TEST_F()
264 HloModuleGroup module_group(TestName()); in TEST_F()
|
| D | hlo_pass_pipeline.h | 82 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override;
95 void MaybeDumpHloAndSaveFilenames(HloModuleGroup& module_group,
122 HloModuleGroup* module_group) { in RunHelper()
|
| D | hlo_pass_interface.h | 43 virtual StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) = 0;
53 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
|
| D | compiler.h | 228 std::unique_ptr<HloModuleGroup> module_group,
232 std::unique_ptr<HloModuleGroup> module_group, in Compile()
262 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
268 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 | hlo_pass_fix.h | 60 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
|
| D | compiler.cc | 48 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| D | compile_only_service.cc | 120 absl::make_unique<HloModuleGroup>(hlo_modules[0]->name(), in CompileAheadOfTime()
|
| D | hlo_runner.cc | 429 auto module_group = absl::make_unique<HloModuleGroup>(std::move(module)); in CreateExecutable()
|
| D | service.cc | 307 absl::make_unique<HloModuleGroup>(module_protos[0]->name()); in BuildExecutables()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.h | 53 std::unique_ptr<HloModuleGroup> module_group,
58 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | compiler.cc | 129 std::unique_ptr<HloModuleGroup> module_group, in Compile()
155 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.h | 129 std::unique_ptr<HloModuleGroup> module_group,
148 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | vectorized_reduce_with_no_vector_registers_test.cc | 74 auto module_group = absl::make_unique<HloModuleGroup>("group"); in TEST_F()
|
| D | cpu_compiler.cc | 191 std::unique_ptr<HloModuleGroup> module_group, in Compile()
804 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 = absl::make_unique<HloModuleGroup>(std::move(hlo_module)); in CompileToAotCompilationResult()
|
| D | llvm_compiler_test.cc | 138 auto module_group = absl::make_unique<HloModuleGroup>("test_module_group"); in TestMultiModuleCompilation()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 276 std::unique_ptr<HloModuleGroup> module_group, in Compile()
351 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.h | 101 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group,
|
| D | gpu_compiler.cc | 978 GpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|