Home
last modified time | relevance | path

Searched refs:HloModule (Results 1 – 25 of 502) sorted by relevance

12345678910>>...21

/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.h82 std::unique_ptr<HloModule> CreateNewUnverifiedModule(
100 HloModule* module);
146 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
151 Literal ExecuteNoHloPasses(std::unique_ptr<HloModule> module,
154 Literal ExecuteAndTransfer(std::unique_ptr<HloModule> module,
163 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments,
168 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments,
191 std::unique_ptr<HloModule> module,
194 const std::function<void(HloModule*)>& reference_preprocessor = nullptr)
200 std::unique_ptr<HloModule> module,
[all …]
Dhlo_test_base.cc75 ProgramShape GetProgramShapeWithLayout(const HloModule& module) { in GetProgramShapeWithLayout()
114 std::unique_ptr<HloModule> HloTestBase::CreateNewUnverifiedModule( in CreateNewUnverifiedModule()
116 return absl::make_unique<HloModule>(name, GetModuleConfigForTest()); in CreateNewUnverifiedModule()
147 HloModule* module) { in RunHloPass()
187 StatusOr<Literal> HloTestBase::Execute(std::unique_ptr<HloModule> module, in Execute()
192 Literal HloTestBase::ExecuteNoHloPasses(std::unique_ptr<HloModule> module, in ExecuteNoHloPasses()
200 Literal HloTestBase::ExecuteAndTransfer(std::unique_ptr<HloModule> module, in ExecuteAndTransfer()
206 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments, in ExecuteReplicated()
219 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments, in ExecuteReplicated()
246 StatusOr<std::unique_ptr<HloModule>> HloTestBase::MakeReferenceModule( in MakeReferenceModule()
[all …]
Dtoken_hlo_test.cc31 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in XLA_TEST_F()
42 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in XLA_TEST_F()
56 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in XLA_TEST_F()
71 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F()
92 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F()
147 std::unique_ptr<HloModule> module, in XLA_TEST_F()
185 std::unique_ptr<HloModule> module, in XLA_TEST_F()
195 std::unique_ptr<HloModule> module, in XLA_TEST_F()
223 std::unique_ptr<HloModule> module, in XLA_TEST_F()
244 std::unique_ptr<HloModule> module, in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser_test.cc103 R"(HloModule module in CreateTestCases()
115 R"(HloModule constant_s32_module in CreateTestCases()
127 R"(HloModule ConstantF32_module in CreateTestCases()
138 R"(HloModule ConstantF32Empty_module in CreateTestCases()
149 R"(HloModule ConstantF32R4Empty_module in CreateTestCases()
160 R"(HloModule Small_3x2x1x1_module in CreateTestCases()
171 R"(HloModule IsFiniteR1F32s_module in CreateTestCases()
183 R"(HloModule ConstantF16_module in CreateTestCases()
194 R"(HloModule BF16 in CreateTestCases()
205 R"(HloModule add_constants_module in CreateTestCases()
[all …]
Ddump.h38 string TimestampFor(const HloModule& module);
41 string FilenameFor(const HloModule& module, absl::string_view prefix,
48 void DumpToFileInDir(const HloModule& module, absl::string_view file_prefix,
53 void DumpToFileInDirOrStdout(const HloModule& module,
69 void DumpHloModuleIfEnabled(const HloModule& module, absl::string_view name);
70 void DumpHloModuleIfEnabled(const HloModule& module,
73 void DumpHloModuleIfEnabled(const HloModule& module,
82 absl::string_view after_pass_name, const HloModule& module);
90 const HloModule& module);
98 void DumpHloSnapshotIfEnabled(const HloModule& module,
[all …]
Dhlo_module_group.h38 explicit HloModuleGroup(std::unique_ptr<HloModule> module);
42 absl::Span<std::unique_ptr<HloModule>> modules);
44 std::vector<std::unique_ptr<HloModule>>&& modules);
47 const std::vector<HloModule*>& modules() const { return module_ptrs_; } in modules()
50 HloModule& module(int index) const { return *module_ptrs_.at(index); } in module()
53 void push_back(std::unique_ptr<HloModule> module);
57 void ReplaceModule(int index, std::unique_ptr<HloModule> module);
61 std::vector<std::unique_ptr<HloModule>> ConsumeModules();
92 std::vector<std::unique_ptr<HloModule>> modules_;
96 std::vector<HloModule*> module_ptrs_;
Dhlo_module_group.cc22 HloModuleGroup::HloModuleGroup(std::unique_ptr<HloModule> module) in HloModuleGroup()
28 absl::Span<std::unique_ptr<HloModule>> modules) in HloModuleGroup()
36 absl::string_view name, std::vector<std::unique_ptr<HloModule>>&& modules) in HloModuleGroup()
43 std::vector<std::unique_ptr<HloModule>> HloModuleGroup::ConsumeModules() { in ConsumeModules()
44 std::vector<std::unique_ptr<HloModule>> ret_modules = std::move(modules_); in ConsumeModules()
55 for (const HloModule* module : modules()) { in ToString()
64 for (const HloModule* module : modules()) { in ToProto()
86 std::vector<std::unique_ptr<HloModule>> modules; in CreateFromProto()
90 std::unique_ptr<HloModule> module, in CreateFromProto()
91 HloModule::CreateFromProto(module_proto, module_configs[i])); in CreateFromProto()
[all …]
Dhlo_runner_interface.cc22 /*static*/ StatusOr<std::unique_ptr<HloModule>>
33 StatusOr<std::unique_ptr<HloModule>> HloProtoToModule( in HloProtoToModule()
36 HloModule::CreateModuleConfigFromProto(proto.hlo_module(), in HloProtoToModule()
39 HloModule::CreateFromProto(proto.hlo_module(), config)); in HloProtoToModule()
45 /*static*/ StatusOr<std::unique_ptr<HloModule>>
54 /*static*/ StatusOr<std::unique_ptr<HloModule>>
63 /*static*/ StatusOr<std::unique_ptr<HloModule>>
74 /*static*/ StatusOr<std::unique_ptr<HloModule>>
83 HloModule::CreateModuleConfigFromProto(module_proto, debug_options)); in ReadModuleFromModuleBinaryProtofile()
85 return HloModule::CreateFromProto(module_proto, module_config); in ReadModuleFromModuleBinaryProtofile()
[all …]
Dhlo_runner_interface.h90 static StatusOr<std::unique_ptr<HloModule>> CreateModuleFromString(
95 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromBinaryProtoFile(
97 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromTextProtoFile(
102 static StatusOr<std::unique_ptr<HloModule>>
108 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromHloTextFile(
114 std::unique_ptr<HloModule> module, bool run_hlo_passes) = 0;
121 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
127 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
132 virtual StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
158 std::unique_ptr<HloModule> module,
[all …]
Dhlo_module.cc44 HloModule::HloModule(const string& name, HloModuleConfig config) in HloModule() function in xla::HloModule
52 Status HloModule::set_schedule(HloSchedule schedule) { in set_schedule()
59 void HloModule::ReplaceEntryComputation(HloComputation* entry_computation) { in ReplaceEntryComputation()
67 HloComputation* HloModule::AddComputationInternal( in AddComputationInternal()
122 HloComputation* HloModule::AddEntryComputation( in AddEntryComputation()
129 HloComputation* HloModule::AddEntryComputationWithLayouts( in AddEntryComputationWithLayouts()
136 Status HloModule::RemoveEmbeddedComputation(HloComputation* to_remove) { in RemoveEmbeddedComputation()
151 HloComputation* HloModule::AddEmbeddedComputation( in AddEmbeddedComputation()
158 void HloModule::ReplaceComputations( in ReplaceComputations()
235 string HloModule::ToString(const HloPrintOptions& options) const { in ToString()
[all …]
Dar_crs_combiner_test.cc41 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
61 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
80 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
101 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
121 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
141 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
163 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
185 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
208 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
244 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
[all …]
Dcopy_insertion.h61 StatusOr<bool> Run(HloModule* module) override;
67 HloModule* module);
79 Status AddSpecialCaseCopies(HloModule* module);
84 HloModule* module);
95 Status AddCopiesToResolveInterference(HloModule* module);
Dwhile_loop_analysis_test.cc52 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
83 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
116 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
151 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
190 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
229 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
264 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
Dhlo_pass_pipeline.cc36 void RecordPassStartMetadata(HloModule& module, const std::string& pass_name, in RecordPassStartMetadata()
47 for (HloModule* module : module_group.modules()) { in RecordPassStartMetadata()
52 Status AttemptRecordPassEndMetadata(HloModule& module, in AttemptRecordPassEndMetadata()
65 void RecordPassEndMetadata(HloModule& module, const std::string& pass_name, in RecordPassEndMetadata()
77 for (HloModule* module : module_group.modules()) { in AttemptRecordPassEndMetadata()
78 for (HloModule* other_module : module_group.modules()) { in AttemptRecordPassEndMetadata()
98 void SetInstructionMetadata(HloModule& module) { in SetInstructionMetadata()
116 for (HloModule* module : module_group.modules()) { in SetInstructionMetadata()
238 HloModule& module, absl::string_view after_pass_name, in MaybeDumpHloAndSaveFilenames()
252 for (HloModule* module : module_group.modules()) { in MaybeDumpHloAndSaveFilenames()
[all …]
Dbatch_dot_simplification_test.cc40 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
64 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
88 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
112 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
136 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
160 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
187 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
208 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
Dhlo_proto_util.cc25 HloProto MakeHloProto(const HloModule& module, in MakeHloProto()
33 HloProto MakeHloProto(const HloModule& module) { in MakeHloProto()
40 StatusOr<std::unique_ptr<HloModule>> CreateModuleFromProto( in CreateModuleFromProto()
44 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloModule> module, in CreateModuleFromProto()
45 HloModule::CreateFromProto(proto, module_config)); in CreateModuleFromProto()
Dhlo_memory_scheduler.h52 HloModule*, const TuplePointsToAnalysis&, const HloAliasAnalysis&,
106 HloModule* module, const TuplePointsToAnalysis& points_to_analysis,
115 HloModule* module, const LogicalBuffer::SizeFunction& size_function,
140 StatusOr<bool> Run(HloModule* module) override;
154 StatusOr<bool> Run(HloModule* module) override;
165 StatusOr<bool> Run(HloModule* module) override;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.h56 StatusOr<std::unique_ptr<HloModule>> RunHloPasses(
57 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
61 std::tuple<std::unique_ptr<HloModule>, std::unique_ptr<BufferAssignment>>>
62 RunHloPassesAndBufferAssignement(std::unique_ptr<HloModule> hlo_module,
66 Status OptimizeHloModule(HloModule* hlo_module,
71 HloModule* hlo_module, se::StreamExecutor* stream_exec,
75 HloModule* hlo_module, se::StreamExecutor* stream_exec,
92 const HloModule* debug_module) = 0;
94 Status PrepareHloModuleForIrEmitting(HloModule* hlo_module);
97 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment_test.cc42 StatusOr<bool> RunParallelTaskAssigner(HloModule* module) { in RunParallelTaskAssigner()
60 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
88 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
105 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
122 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
167 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
188 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
202 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
Dcpu_compiler.h133 StatusOr<std::unique_ptr<HloModule>> RunHloPasses(
134 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
138 std::tuple<std::unique_ptr<HloModule>, std::unique_ptr<BufferAssignment>>>
139 RunHloPassesAndBufferAssignement(std::unique_ptr<HloModule> module,
144 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
161 Status RunHloPasses(HloModule* module, bool is_aot_compile,
166 HloModule* module, bool /*is_aot_compile*/,
171 HloModule* module, bool is_aot_compile,
Dparallel_task_assignment.h43 HloModule* module,
82 StatusOr<bool> Run(HloModule* module) override;
90 bool AssignParallelTasks(HloModule* module,
93 HloModule* module, HloComputation* computation,
98 void ComputeTargetParallelTasks(HloModule* module,
/external/tensorflow/tensorflow/compiler/xla/tools/
Dprepare_reference_module.cc33 StatusOr<std::unique_ptr<HloModule>> PrepareReferenceModule( in PrepareReferenceModule()
34 const HloModule& test_module, in PrepareReferenceModule()
37 const std::function<Status(const HloModule&, in PrepareReferenceModule() argument
39 HloModule*)>& module_modifier_hook) { in PrepareReferenceModule()
51 std::unique_ptr<HloModule> reference_module = in PrepareReferenceModule()
Dprepare_reference_module.h35 StatusOr<std::unique_ptr<HloModule>> PrepareReferenceModule(
36 const HloModule& test_module,
39 const std::function<Status(const HloModule&,
41 HloModule*)>& module_modifier_hook = {});
Dhlo_extractor.cc43 const HloModule& old_module, in ExtractionVisitor()
46 module_(absl::make_unique<HloModule>("extracted", config_)), in ExtractionVisitor()
97 HloModule* module() { return module_.get(); } in module()
99 std::unique_ptr<HloModule> ConsumeModule() { return std::move(module_); } in ConsumeModule()
102 const HloModule& old_module_;
104 std::unique_ptr<HloModule> module_;
137 std::unique_ptr<HloModule> ExtractModule(HloInstruction* instruction, in ExtractModule()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir4 // CHECK: HloModule
17 // CHECK: HloModule
46 // CHECK: HloModule
74 // CHECK: HloModule
91 // CHECK: HloModule
106 // CHECK: HloModule
129 // CHECK: HloModule
141 // CHECK: HloModule
151 // CHECK: HloModule
165 // CHECK: HloModule
[all …]

12345678910>>...21