Home
last modified time | relevance | path

Searched refs:hlo_module (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_kernel_tiling_test.cc63 auto hlo_module = in TEST_F() local
65 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
89 auto hlo_module = in TEST_F() local
91 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
116 auto hlo_module = in TEST_F() local
118 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
150 auto hlo_module = in TEST_F() local
152 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
185 auto hlo_module = in TEST_F() local
187 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
[all …]
Dgpu_unrolling_test.cc51 auto hlo_module = ParseHloString(kAddModule, config).ValueOrDie(); in TEST_F() local
53 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
70 auto hlo_module = ParseHloString(kAddModule, config).ValueOrDie(); in TEST_F() local
72 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
89 auto hlo_module = ParseHloString(kAddModule, config).ValueOrDie(); in TEST_F() local
91 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
120 auto hlo_module = ParseHloString(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() local
122 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
163 auto hlo_module = in TEST_F() local
166 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
Dgpu_ldg_test.cc51 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
52 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
54 CompileAndVerifyPtx(std::move(hlo_module), R"( in TEST_F()
76 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
77 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
79 CompileAndVerifyPtx(std::move(hlo_module), R"( in TEST_F()
98 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
111 hlo_module->AddEmbeddedComputation(embedded_builder.Build()); in TEST_F()
129 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
131 CompileAndVerifyPtx(std::move(hlo_module), R"( in TEST_F()
Dgpu_ftz_test.cc42 auto hlo_module = CreateNewUnverifiedModuleWithFTZ(ftz_); in CreateBinaryOpModule() local
43 hlo_module->AddEntryComputation(builder.Build()); in CreateBinaryOpModule()
44 return hlo_module; in CreateBinaryOpModule()
57 auto hlo_module = CreateNewUnverifiedModuleWithFTZ(ftz_); in CreateUnaryOpModule() local
58 hlo_module->AddEntryComputation(builder.Build()); in CreateUnaryOpModule()
59 return hlo_module; in CreateUnaryOpModule()
Dgpu_copy_test.cc48 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
49 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
52 CompileAndVerifyIr(std::move(hlo_module), "; CHECK-NOT: define void @_copy", in TEST_F()
Dgpu_noalias_test.cc50 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
51 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
53 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
Dgpu_index_test.cc55 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
56 hlo_module->AddEntryComputation(builder.Build()); in TEST_F()
59 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc45 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
46 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
47 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
61 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
62 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
63 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
84 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
85 hlo_module->AddEntryComputation(builder.Build()); in XLA_TEST_F()
86 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
105 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
[all …]
Dfusion_test.cc92 auto hlo_module = CreateNewVerifiedModule(); in TestElementwise2D() local
125 hlo_module->AddEntryComputation(builder.Build()) in TestElementwise2D()
131 auto actual = ExecuteAndTransfer(std::move(hlo_module), {}); in TestElementwise2D()
202 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
234 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
242 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
250 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
262 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
268 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
285 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
[all …]
Dcodegen_test_base.cc21 std::unique_ptr<HloModule> hlo_module) { in CompileToExecutable() argument
22 TF_ASSIGN_OR_RETURN(hlo_module, backend().compiler()->RunHloPasses( in CompileToExecutable()
23 std::move(hlo_module), in CompileToExecutable()
26 return backend().compiler()->RunBackend(std::move(hlo_module), in CompileToExecutable()
33 std::unique_ptr<HloModule> hlo_module, in CompileToAotCompilationResult() argument
35 auto module_group = absl::make_unique<HloModuleGroup>(std::move(hlo_module)); in CompileToAotCompilationResult()
Dllvm_compiler_test.cc71 auto hlo_module = CreateNewUnverifiedModule(); in TestCompilerHooks() local
72 hlo_module->AddEntryComputation(builder.Build()); in TestCompilerHooks()
78 ->RunBackend(std::move(hlo_module), in TestCompilerHooks()
93 std::unique_ptr<HloModule> hlo_module = CreateNewUnverifiedModule(); in TestMultiModuleCompilation() local
94 hlo_module->AddEntryComputation(builder.Build()); in TestMultiModuleCompilation()
97 module_group->push_back(hlo_module->Clone()); in TestMultiModuleCompilation()
98 module_group->push_back(std::move(hlo_module)); in TestMultiModuleCompilation()
Dllvm_irgen_test_base.cc51 std::unique_ptr<HloModule> hlo_module, const string& pattern, in CompileAndVerifyIr() argument
54 Status status = CompileToExecutable(std::move(hlo_module)).status(); in CompileAndVerifyIr()
74 std::unique_ptr<HloModule> hlo_module, const AotCompilationOptions& options, in CompileAheadOfTimeAndVerifyIr() argument
78 CompileToAotCompilationResult(std::move(hlo_module), options).status(); in CompileAheadOfTimeAndVerifyIr()
/external/tensorflow/tensorflow/compiler/xla/service/
Dmap_inliner_test.cc62 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
63 hlo_module->AddEmbeddedComputation(std::move(max_f32)); in TEST_F()
64 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
67 EXPECT_TRUE(inliner.Run(hlo_module.get()).ValueOrDie()); in TEST_F()
68 EXPECT_THAT(hlo_module->entry_computation()->root_instruction(), in TEST_F()
72 auto result = ExecuteAndTransfer(hlo_module->Clone(), {}); in TEST_F()
96 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
97 hlo_module->AddEmbeddedComputation(std::move(const2_f32)); in TEST_F()
98 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
99 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
[all …]
Dhlo_cost_analysis_test.cc146 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
149 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
175 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
178 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
204 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
207 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
227 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
230 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
248 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
251 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
[all …]
Dhlo_proto_util.cc57 if (!hlo_proto.hlo_module().has_host_program_shape()) { in EntryComputationParameterShapes()
62 const auto& program_shape = hlo_proto.hlo_module().host_program_shape(); in EntryComputationParameterShapes()
74 if (!hlo_proto.hlo_module().has_host_program_shape()) { in EntryComputationOutputShape()
77 if (!hlo_proto.hlo_module().host_program_shape().has_result()) { in EntryComputationOutputShape()
81 return &hlo_proto.hlo_module().host_program_shape().result(); in EntryComputationOutputShape()
Dpattern_matcher_test.cc36 TF_ASSERT_OK_AND_ASSIGN(auto hlo_module, ParseHloString(kModuleStr)); in TEST()
44 hlo_module->entry_computation()->root_instruction(), in TEST()
58 EXPECT_TRUE(Match(hlo_module->entry_computation()->root_instruction(), in TEST()
61 EXPECT_FALSE(Match(hlo_module->entry_computation()->root_instruction(), in TEST()
64 EXPECT_FALSE(Match(hlo_module->entry_computation()->root_instruction(), in TEST()
189 TF_ASSERT_OK_AND_ASSIGN(auto hlo_module, ParseHloString(kModuleStr)); in TEST()
191 auto* root = hlo_module->entry_computation()->root_instruction(); in TEST()
208 TF_ASSERT_OK_AND_ASSIGN(auto hlo_module, ParseHloString(kModuleStr)); in TEST()
210 auto* root = hlo_module->entry_computation()->root_instruction(); in TEST()
221 TF_ASSERT_OK_AND_ASSIGN(auto hlo_module, ParseHloString(kModuleStr)); in TEST()
[all …]
Dhlo_execution_profile_test.cc32 auto hlo_module = ParseHloString(R"( in TEST_F() local
42 hlo_module->entry_computation()->root_instruction(); in TEST_F()
55 HloProfileIndexMap profile_index_map(*hlo_module); in TEST_F()
58 hlo_module->entry_computation()->name()); in TEST_F()
Dcompile_only_service.cc89 std::unique_ptr<HloModule> hlo_module, in CompileAheadOfTime()
91 DumpHloModuleIfEnabled(*hlo_module, "before_optimizations"); in CompileAheadOfTime()
92 hlo_modules.push_back(std::move(hlo_module)); in CompileAheadOfTime()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor_test.cc39 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> hlo_module, in TEST_F()
44 ExtractModule(FindInstruction(hlo_module.get(), "exp")); in TEST_F()
51 ExtractModule(FindInstruction(hlo_module.get(), "exp"), /*height=*/0); in TEST_F()
58 FindInstruction(hlo_module.get(), "negate"), /*height=*/0); in TEST_F()
77 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> hlo_module, in TEST_F()
82 ExtractModule(FindInstruction(hlo_module.get(), "exp")); in TEST_F()
89 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/0); in TEST_F()
95 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/1); in TEST_F()
102 ExtractModule(FindInstruction(hlo_module.get(), "add"), /*height=*/2); in TEST_F()
121 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> hlo_module, in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc80 Status InterpreterCompiler::RunHloOptimization(HloModule* hlo_module) { in RunHloOptimization() argument
87 hlo_module->mutable_entry_computation_layout(), in RunHloOptimization()
91 &pipeline, hlo_module->config().debug_options(), in RunHloOptimization()
94 return pipeline.Run(hlo_module).status(); in RunHloOptimization()
98 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses() argument
100 VLOG(1) << "Run hlo passes on graph " << hlo_module->name(); in RunHloPasses()
101 TF_RETURN_IF_ERROR(RunHloOptimization(hlo_module.get())); in RunHloPasses()
102 return std::move(hlo_module); in RunHloPasses()
113 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend() argument
117 VLOG(1) << "Run backend " << hlo_module->name(); in RunBackend()
[all …]
Dcompiler.h47 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec,
55 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec,
76 Status RunHloOptimization(HloModule* hlo_module);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_noalias_test.cc59 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
60 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
64 hlo_module.get(), in TEST_F()
65 absl::make_unique<DependencyHloOrdering>(hlo_module.get()), in TEST_F()
71 llvm_ir::AliasAnalysis aa(*hlo_module, *status_or_buffer_assn.ValueOrDie(), in TEST_F()
Dcpu_eigen_dot_operation_test.cc52 auto hlo_module = CreateNewVerifiedModule(); in CompileAndCheck() local
53 hlo_module->AddEntryComputation(std::move(entry_computation)); in CompileAndCheck()
55 CompileAheadOfTimeAndVerifyIr(std::move(hlo_module), options, in CompileAndCheck()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc179 Status OptimizeHloModule(HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloModule() argument
189 &pipeline, hlo_module->config().debug_options(), in OptimizeHloModule()
218 if (hlo_module->config().debug_options().xla_gpu_use_cudnn_batchnorm()) { in OptimizeHloModule()
264 TF_RETURN_IF_ERROR(pipeline.Run(hlo_module).status()); in OptimizeHloModule()
287 TF_RETURN_IF_ERROR(pipeline.Run(hlo_module).status()); in OptimizeHloModule()
299 hlo_module->mutable_entry_computation_layout(), in OptimizeHloModule()
301 TF_RETURN_IF_ERROR(pipeline.Run(hlo_module).status()); in OptimizeHloModule()
351 TF_RETURN_IF_ERROR(pipeline.Run(hlo_module).status()); in OptimizeHloModule()
372 TF_RETURN_IF_ERROR(fusion.Run(hlo_module).status()); in OptimizeHloModule()
381 &reduce_pipeline, hlo_module->config().debug_options(), in OptimizeHloModule()
[all …]
Dir_emitter_context.h33 IrEmitterContext(const HloModule* hlo_module, in IrEmitterContext() argument
37 : hlo_module_(hlo_module), in IrEmitterContext()
46 const HloModule& hlo_module() const { return *hlo_module_; } in hlo_module() function

123