| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 82 std::unique_ptr<HloModule> CreateNewUnverifiedModule(
106 HloModule* module);
108 HloModule* module) { in RunHloPass()
112 HloModule* module) { in RunHloPass()
149 std::function<void(HloModule*)> after_pass_checks = nullptr);
170 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
175 Literal ExecuteNoHloPasses(std::unique_ptr<HloModule> module,
178 Literal ExecuteAndTransfer(std::unique_ptr<HloModule> module,
187 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments,
192 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments,
[all …]
|
| D | hlo_test_base.cc | 63 ProgramShape GetProgramShapeWithLayout(const HloModule& module) { in GetProgramShapeWithLayout()
112 std::unique_ptr<HloModule> HloTestBase::CreateNewUnverifiedModule( in CreateNewUnverifiedModule()
114 return std::make_unique<HloModule>(name, GetModuleConfigForTest()); in CreateNewUnverifiedModule()
150 HloModule* module, std::unique_ptr<HloComputation> computation) { in AddEntryComputationAndUpdateEntryComputationLayout()
156 void HloTestBase::UpdateEntryComputationLayout(HloModule* module) { in UpdateEntryComputationLayout()
163 HloModule* module) { in RunHloPass()
212 std::function<void(HloModule*)> after_pass_checks) { in RunAndFilecheckHloRewrite()
230 StatusOr<Literal> HloTestBase::Execute(std::unique_ptr<HloModule> module, in Execute()
235 Literal HloTestBase::ExecuteNoHloPasses(std::unique_ptr<HloModule> module, in ExecuteNoHloPasses()
243 Literal HloTestBase::ExecuteAndTransfer(std::unique_ptr<HloModule> module, in ExecuteAndTransfer()
[all …]
|
| D | token_hlo_test.cc | 30 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in XLA_TEST_F()
41 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in XLA_TEST_F()
55 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in XLA_TEST_F()
70 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F()
91 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F()
146 std::unique_ptr<HloModule> module, in XLA_TEST_F()
184 std::unique_ptr<HloModule> module, in XLA_TEST_F()
194 std::unique_ptr<HloModule> module, in XLA_TEST_F()
222 std::unique_ptr<HloModule> module, in XLA_TEST_F()
243 std::unique_ptr<HloModule> module, in XLA_TEST_F()
[all …]
|
| D | llvm_irgen_test_base.cc | 50 std::unique_ptr<HloModule> hlo_module, const std::string& pattern, in CompileAndVerifyIr()
67 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in CompileAndVerifyIr()
73 std::unique_ptr<HloModule> hlo_module, const AotCompilationOptions& options, in CompileAheadOfTimeAndVerifyIr()
89 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> optimized_module, in MatchOptimizedHlo()
99 StatusOr<std::unique_ptr<HloModule>> LlvmIrGenTestBase::GetOptimizedModule( in GetOptimizedModule()
102 std::unique_ptr<HloModule> module, in GetOptimizedModule()
109 StatusOr<std::unique_ptr<HloModule>> LlvmIrGenTestBase::GetOptimizedModule( in GetOptimizedModule()
110 std::unique_ptr<HloModule> hlo_module) { in GetOptimizedModule()
|
| D | llvm_irgen_test_base.h | 38 void CompileAndVerifyIr(std::unique_ptr<HloModule> hlo_module,
56 void CompileAheadOfTimeAndVerifyIr(std::unique_ptr<HloModule> hlo_module,
73 StatusOr<std::unique_ptr<HloModule>> GetOptimizedModule(
76 StatusOr<std::unique_ptr<HloModule>> GetOptimizedModule(
77 std::unique_ptr<HloModule> hlo_module);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dump.h | 39 std::string TimestampFor(const HloModule& module);
42 std::string FilenameFor(const HloModule& module, absl::string_view prefix,
49 void DumpToFileInDir(const HloModule& module, absl::string_view file_prefix,
56 void DumpToFileInDirOrStdout(const HloModule& module,
74 void DumpToFileInDirOrStdout(const HloModule& module,
87 void DumpPerModuleProtobufToFile(const HloModule& module,
98 void DumpHloModuleIfEnabled(const HloModule& module, absl::string_view name);
99 void DumpHloModuleIfEnabled(const HloModule& module,
102 void DumpHloModuleIfEnabled(const HloModule& module,
111 absl::string_view after_pass_name, const HloModule& module);
[all …]
|
| D | hlo_parser_test.cc | 126 R"(HloModule module, entry_computation_layout={()->pred[2,3]{1,0}} in CreateTestCases()
138 R"(HloModule constant_s32_module, entry_computation_layout={()->s32[]} in CreateTestCases()
150 R"(HloModule ConstantF32_module, entry_computation_layout={()->f32[]} in CreateTestCases()
161 R"(HloModule ConstantF32Empty_module, entry_computation_layout={()->f32[0]{0}} in CreateTestCases()
172 R"(HloModule ConstantF32R4Empty_module, entry_computation_layout={()->f32[2,0,4,3]{3,2,1,0}} in CreateTestCases()
183 R"(HloModule Small_3x2x1x1_module, entry_computation_layout={()->f32[3,2,1,1]{3,2,1,0}} in CreateTestCases()
194 R"(HloModule IsFiniteR1F32s_module, entry_computation_layout={()->pred[6]{0}} in CreateTestCases()
206 R"(HloModule ConstantF16_module, entry_computation_layout={()->f16[]} in CreateTestCases()
217 R"(HloModule BF16, entry_computation_layout={()->bf16[]} in CreateTestCases()
228 R"(HloModule add_constants_module, entry_computation_layout={()->f32[]} in CreateTestCases()
[all …]
|
| 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 | 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();
103 std::vector<std::unique_ptr<HloModule>> modules_;
107 std::vector<HloModule*> module_ptrs_;
|
| D | hlo_runner_interface.cc | 22 /*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 …]
|
| D | hlo_runner_interface.h | 90 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,
156 std::unique_ptr<HloModule> module,
[all …]
|
| D | ar_crs_combiner_test.cc | 42 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
62 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
81 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
102 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
122 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
142 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
164 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()
209 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
245 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
[all …]
|
| D | hlo_module.cc | 57 HloModule::HloModule(const std::string& name, HloModuleConfig config) in HloModule() function in xla::HloModule
58 : HloModule(name, config, std::make_unique<CompilationEnvironments>()) {} in HloModule()
60 Status HloModule::set_schedule(HloSchedule schedule) { in set_schedule()
67 void HloModule::ReplaceEntryComputation(HloComputation* entry_computation) { in ReplaceEntryComputation()
75 HloComputation* HloModule::AddComputationInternal( in AddComputationInternal()
130 HloComputation* HloModule::AddEntryComputation( in AddEntryComputation()
137 HloComputation* HloModule::AddEntryComputationWithLayouts( in AddEntryComputationWithLayouts()
144 Status HloModule::RemoveEmbeddedComputation(HloComputation* to_remove) { in RemoveEmbeddedComputation()
159 HloComputation* HloModule::AddEmbeddedComputation( in AddEmbeddedComputation()
166 void HloModule::ReplaceComputations( in ReplaceComputations()
[all …]
|
| D | dot_merger_test.cc | 50 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
79 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
102 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
145 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
184 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
203 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
226 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
249 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
272 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
313 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
[all …]
|
| D | all_gather_combiner_test.cc | 42 int64_t AllGatherCount(const HloModule& module) { in AllGatherCount()
72 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
108 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()
167 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()
215 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
246 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
286 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
310 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
|
| D | while_loop_analysis_test.cc | 52 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()
|
| D | convert_mover_test.cc | 60 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()
95 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
114 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
133 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
153 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
176 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
192 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
214 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
234 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
[all …]
|
| D | batch_dot_simplification_test.cc | 41 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
65 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
89 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
113 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
137 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
161 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()
209 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
|
| D | hlo_pass_interface.h | 46 explicit RunState(HloModule* module) in RunState()
86 StatusOr<bool> Run(HloModule* module) { in Run()
90 HloModule* module,
101 HloModule* module, RunState* run_state, in RunOnChangedComputations()
154 for (HloModule* module : module_group->modules()) { in RunOnModuleGroup()
174 StatusOr<bool> Run(HloModule* module, in Run()
|
| D | hlo_pass_pipeline.cc | 38 void RecordPassStartMetadata(HloModule& module, const std::string& pass_name, in RecordPassStartMetadata()
49 for (HloModule* module : module_group.modules()) { in RecordPassStartMetadata()
54 Status AttemptRecordPassEndMetadata(HloModule& module, in AttemptRecordPassEndMetadata()
67 void RecordPassEndMetadata(HloModule& module, const std::string& pass_name, in RecordPassEndMetadata()
79 for (HloModule* module : module_group.modules()) { in AttemptRecordPassEndMetadata()
80 for (HloModule* other_module : module_group.modules()) { in AttemptRecordPassEndMetadata()
100 void SetInstructionMetadata(HloModule& module) { in SetInstructionMetadata()
118 for (HloModule* module : module_group.modules()) { in SetInstructionMetadata()
251 HloModule& module, absl::string_view after_pass_name, in MaybeDumpHloAndSaveFilenames()
265 for (HloModule* module : module_group.modules()) { in MaybeDumpHloAndSaveFilenames()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment_test.cc | 43 StatusOr<bool> RunParallelTaskAssigner(HloModule* module) { in RunParallelTaskAssigner()
61 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
89 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
106 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
123 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
168 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
189 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
203 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
|
| D | cpu_compiler.h | 138 StatusOr<std::unique_ptr<HloModule>> RunHloPasses(
139 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
143 const HloModule* module) override;
146 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
163 Status RunHloPasses(HloModule* module, bool is_aot_compile,
169 HloModule* module, bool /*is_aot_compile*/,
175 HloModule* module, bool is_aot_compile,
179 std::unique_ptr<HloModule> module);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.h | 86 StatusOr<std::unique_ptr<HloModule>> RunHloPasses(
87 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
91 const HloModule* hlo_module) override;
96 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec,
107 const HloModule* debug_module);
122 HloModule* hlo_module, se::StreamExecutor* stream_exec,
126 Status OptimizeHloModule(HloModule* hlo_module,
131 HloModule* hlo_module, se::StreamExecutor* stream_exec,
146 const HloModule* debug_module) = 0;
148 Status PrepareHloModuleForIrEmitting(HloModule* hlo_module);
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | prepare_reference_module.cc | 33 StatusOr<std::unique_ptr<HloModule>> PrepareReferenceModule( in PrepareReferenceModule()
34 const HloModule& test_module, HloRunnerInterface* test_runner, in PrepareReferenceModule()
36 const std::function<Status(const HloModule&, HloRunnerInterface*, in PrepareReferenceModule() argument
37 HloModule*)>& module_modifier_hook) { in PrepareReferenceModule()
49 std::unique_ptr<HloModule> reference_module = in PrepareReferenceModule()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 4 // CHECK: HloModule
16 // CHECK: HloModule
29 // CHECK: HloModule
59 // CHECK: HloModule
79 // CHECK: HloModule
108 // CHECK: HloModule
136 // CHECK: HloModule
166 // CHECK: HloModule
196 // CHECK: HloModule
219 // CHECK: HloModule
[all …]
|