| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 45 class VerifiedHloModule : public HloModule {
51 : HloModule(name, config), in VerifiedHloModule()
107 std::unique_ptr<HloModule> CreateNewUnverifiedModule(
124 HloModule* module);
165 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
170 Literal ExecuteNoHloPasses(std::unique_ptr<HloModule> module,
173 Literal ExecuteAndTransfer(std::unique_ptr<HloModule> module,
182 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments,
187 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments,
203 std::unique_ptr<HloModule> module,
[all …]
|
| D | hlo_test_base.cc | 75 ProgramShape GetProgramShapeWithLayout(const HloModule& module) { in GetProgramShapeWithLayout()
133 std::unique_ptr<HloModule> HloTestBase::CreateNewUnverifiedModule( in CreateNewUnverifiedModule()
135 return absl::make_unique<HloModule>(name, GetModuleConfigForTest()); in CreateNewUnverifiedModule()
160 HloModule* module) { in RunHloPass()
190 StatusOr<Literal> HloTestBase::Execute(std::unique_ptr<HloModule> module, in Execute()
195 Literal HloTestBase::ExecuteNoHloPasses(std::unique_ptr<HloModule> module, in ExecuteNoHloPasses()
203 Literal HloTestBase::ExecuteAndTransfer(std::unique_ptr<HloModule> module, in ExecuteAndTransfer()
209 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments, in ExecuteReplicated()
221 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments, in ExecuteReplicated()
234 StatusOr<std::unique_ptr<HloModule>> HloTestBase::MakeReferenceModule( in MakeReferenceModule()
[all …]
|
| D | token_hlo_test.cc | 32 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F()
43 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F()
57 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F()
72 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F()
93 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F()
148 std::unique_ptr<HloModule> module, in XLA_TEST_F()
186 std::unique_ptr<HloModule> module, in XLA_TEST_F()
196 std::unique_ptr<HloModule> module, in XLA_TEST_F()
224 std::unique_ptr<HloModule> module, in XLA_TEST_F()
245 std::unique_ptr<HloModule> module, in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser_test.cc | 95 R"(HloModule module in CreateTestCases()
107 R"(HloModule constant_s32_module in CreateTestCases()
119 R"(HloModule ConstantF32_module in CreateTestCases()
130 R"(HloModule ConstantF32Empty_module in CreateTestCases()
141 R"(HloModule ConstantF32R4Empty_module in CreateTestCases()
152 R"(HloModule Small_3x2x1x1_module in CreateTestCases()
163 R"(HloModule IsFiniteR1F32s_module in CreateTestCases()
175 R"(HloModule ConstantF16_module in CreateTestCases()
186 R"(HloModule BF16 in CreateTestCases()
197 R"(HloModule add_constants_module in CreateTestCases()
[all …]
|
| D | hlo_module_group.h | 38 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();
83 std::vector<std::unique_ptr<HloModule>> modules_;
87 std::vector<HloModule*> module_ptrs_;
|
| D | hlo_module_group.cc | 20 HloModuleGroup::HloModuleGroup(std::unique_ptr<HloModule> module) in HloModuleGroup()
26 absl::Span<std::unique_ptr<HloModule>> modules) in HloModuleGroup()
34 absl::string_view name, std::vector<std::unique_ptr<HloModule>>&& modules) in HloModuleGroup()
41 std::vector<std::unique_ptr<HloModule>> HloModuleGroup::ConsumeModules() { in ConsumeModules()
42 std::vector<std::unique_ptr<HloModule>> ret_modules = std::move(modules_); in ConsumeModules()
53 for (const HloModule* module : modules()) { in ToString()
62 for (const HloModule* module : modules()) { in ToProto()
76 std::vector<std::unique_ptr<HloModule>> modules; in CreateFromProto()
80 std::unique_ptr<HloModule> module, in CreateFromProto()
81 HloModule::CreateFromProto(module_proto, module_configs[i])); in CreateFromProto()
[all …]
|
| D | dump.h | 40 void DumpToFileInDir(const HloModule& module, absl::string_view file_suffix,
45 void DumpToFileInDirOrStdout(const HloModule& module,
55 void DumpHloModuleIfEnabled(const HloModule& module, absl::string_view name);
56 void DumpHloModuleIfEnabled(const HloModule& module,
59 void DumpHloModuleIfEnabled(const HloModule& module,
68 const HloModule& module);
76 const HloModule& module);
84 void DumpHloSnapshotIfEnabled(const HloModule& module,
94 inline bool DumpingEnabledForHloModule(const HloModule& module) { in DumpingEnabledForHloModule()
|
| D | copy_insertion.h | 61 StatusOr<bool> Run(HloModule* module) override;
72 static StatusOr<bool> AddCopiesForBufferAssignment(HloModule* module);
78 HloModule* module);
90 Status AddSpecialCaseCopies(HloModule* module);
95 HloModule* module);
100 HloModule* module);
103 Status AddCopiesToResolveInterference(HloModule* module);
|
| D | hlo_runner.h | 89 static StatusOr<std::unique_ptr<HloModule>> CreateModuleFromString(
94 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromBinaryProtoFile(
96 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromTextProtoFile(
101 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromHloTextFile(
117 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
122 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
138 std::unique_ptr<HloModule> module,
143 std::unique_ptr<HloModule> module,
163 std::unique_ptr<HloModule> module, bool run_hlo_passes);
173 std::unique_ptr<HloModule> module,
[all …]
|
| D | hlo_module.cc | 39 HloModule::HloModule(const string& name, const HloModuleConfig& config) in HloModule() function in xla::HloModule
44 Status HloModule::set_schedule(HloSchedule schedule) { in set_schedule()
51 HloComputation* HloModule::AddComputationInternal( in AddComputationInternal()
103 HloComputation* HloModule::AddEntryComputation( in AddEntryComputation()
109 Status HloModule::RemoveEmbeddedComputation(HloComputation* to_remove) { in RemoveEmbeddedComputation()
119 HloComputation* HloModule::AddEmbeddedComputation( in AddEmbeddedComputation()
125 void HloModule::ReplaceComputations( in ReplaceComputations()
200 string HloModule::ToString(const HloPrintOptions& options) const { in ToString()
223 HloModuleProto HloModule::ToProto() const { in ToProto()
244 Status HloModule::CheckUniqueNamesAndIdsForComputationsAndInstructions() const { in CheckUniqueNamesAndIdsForComputationsAndInstructions()
[all …]
|
| D | ar_crs_combiner_test.cc | 41 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 …]
|
| D | while_loop_constant_sinking_test.cc | 57 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
97 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
138 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
186 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
228 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
274 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
314 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
357 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
408 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
|
| D | batch_dot_simplification_test.cc | 40 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()
|
| D | hlo_module_group_test.cc | 47 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
64 std::vector<std::unique_ptr<HloModule>> modules = group.ConsumeModules(); in TEST_F()
86 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_0, in TEST_F()
88 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_1, in TEST_F()
90 std::vector<std::unique_ptr<HloModule>> modules; in TEST_F()
125 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_0, in TEST_F()
127 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module_1, in TEST_F()
180 std::unique_ptr<HloModule> module, in TEST_F()
|
| D | hlo_proto_util.cc | 25 HloProto MakeHloProto(const HloModule& module, in MakeHloProto()
33 HloProto MakeHloProto(const HloModule& module) { in MakeHloProto()
40 StatusOr<std::unique_ptr<HloModule>> CreateModuleFromProto( in CreateModuleFromProto()
43 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloModule> module, in CreateModuleFromProto()
44 HloModule::CreateFromProto(proto, module_config)); in CreateModuleFromProto()
|
| D | bfloat16_propagation.h | 73 StatusOr<bool> Run(HloModule* module) override;
129 void ResolveInconsistencyOfAliasingBuffers(HloModule* module);
151 Status ResolveInconsistentFusions(HloModule* module);
155 Status ResolveConvertedConstants(HloModule* module);
159 Status SkipNoopConversions(HloModule* module);
|
| D | hlo_schedule.h | 30 class HloModule; variable
83 explicit HloSchedule(const HloModule* module) : module_(module) {} in HloSchedule()
86 static StatusOr<HloSchedule> CreateFromProto(const HloModule* module,
140 const HloModule* module() const { return module_; } in module()
146 const HloModule* module_;
|
| D | hlo_module_test.cc | 215 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
219 std::unique_ptr<HloModule> module_copy, in TEST_F()
220 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
237 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
241 std::unique_ptr<HloModule> module_copy, in TEST_F()
242 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
274 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
308 std::unique_ptr<HloModule> module_copy, in TEST_F()
309 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
|
| D | while_util_test.cc | 29 StatusOr<std::unique_ptr<HloModule>> GetParsedModule( in GetParsedModule()
53 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloModule> module, in GetParsedModule()
69 std::unique_ptr<HloModule> module, in TEST()
100 std::unique_ptr<HloModule> module, in TEST()
154 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST()
195 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST()
|
| D | hlo_module_group_metadata.h | 113 explicit HloModuleGroupMetadata(absl::Span<HloModule* const> modules) in HloModuleGroupMetadata()
120 absl::Span<HloModule* const> modules);
166 int64 GetModuleId(const HloModule* module) const;
210 TuplePointsToAnalysis* points_to_analysis(HloModule* module) const { in points_to_analysis()
283 const std::vector<HloModule*> modules_;
285 absl::flat_hash_map<HloModule*, std::unique_ptr<TuplePointsToAnalysis>>
|
| D | hlo_clone_context.h | 28 class HloModule; variable
37 explicit HloCloneContext(HloModule* module, const string& suffix = "")
40 HloModule* module() const { return module_; } in module()
87 HloModule* module_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.h | 43 HloModule* module,
82 StatusOr<bool> Run(HloModule* module) override;
90 bool AssignParallelTasks(HloModule* module,
93 HloModule* module, HloComputation* computation,
98 void ComputeTargetParallelTasks(HloModule* module,
|
| D | parallel_task_assignment_test.cc | 42 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()
|
| D | cpu_compiler.h | 136 StatusOr<std::unique_ptr<HloModule>> RunHloPasses(
137 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
141 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
158 Status RunHloPasses(HloModule* module, bool is_aot_compile,
163 HloModule* module, bool /*is_aot_compile*/,
168 HloModule* module, bool is_aot_compile,
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_extractor.cc | 42 const HloModule& old_module, in ExtractionVisitor()
45 module_(absl::make_unique<HloModule>("extracted", config_)), in ExtractionVisitor()
96 HloModule* module() { return module_.get(); } in module()
98 std::unique_ptr<HloModule> ConsumeModule() { return std::move(module_); } in ConsumeModule()
101 const HloModule& old_module_;
103 std::unique_ptr<HloModule> module_;
136 std::unique_ptr<HloModule> ExtractModule(HloInstruction* instruction, in ExtractModule()
|