Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_kernel_tiling_test.cc63 auto hlo_module = in TEST_F() local
78 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
97 auto hlo_module = in TEST_F() local
110 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
146 auto hlo_module = in TEST_F() local
159 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
186 auto hlo_module = in TEST_F() local
199 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
227 auto hlo_module = in TEST_F() local
240 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
[all …]
Dgpu_unrolling_test.cc51 auto hlo_module = in TEST_F() local
54 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
71 auto hlo_module = in TEST_F() local
74 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
91 auto hlo_module = in TEST_F() local
94 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
123 auto hlo_module = in TEST_F() local
126 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
154 auto hlo_module = in TEST_F() local
171 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
[all …]
Dgpu_ldg_test.cc56 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
57 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
59 CompileAndOptionallyVerifyPtx(std::move(hlo_module), R"( in TEST_F()
86 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
87 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
89 CompileAndOptionallyVerifyPtx(std::move(hlo_module), R"( in TEST_F()
113 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
126 hlo_module->AddEmbeddedComputation(embedded_builder.Build()); in TEST_F()
144 hlo_module->AddEntryComputation(std::move(computation)); in TEST_F()
146 CompileAndOptionallyVerifyPtx(std::move(hlo_module), R"( in TEST_F()
Dgpu_input_fusible_slice_test.cc63 auto hlo_module = in TEST_F() local
76 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
105 auto hlo_module = in TEST_F() local
118 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
152 auto hlo_module = in TEST_F() local
165 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
Dgpu_ftz_test.cc45 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateBinaryOpModule() local
46 hlo_module->AddEntryComputation(builder.Build()); in CreateBinaryOpModule()
47 return hlo_module; in CreateBinaryOpModule()
60 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateUnaryOpModule() local
61 hlo_module->AddEntryComputation(builder.Build()); in CreateUnaryOpModule()
62 return hlo_module; in CreateUnaryOpModule()
Dgpu_dyn_shape_test.cc35 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
36 hlo_module->AddEntryComputation(builder.Build()); in TEST_F()
38 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
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()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc45 auto hlo_module = CreateNewVerifiedModule(); 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 = CreateNewVerifiedModule(); 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 = CreateNewVerifiedModule(); 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 = CreateNewVerifiedModule(); in XLA_TEST_F() local
[all …]
Dcpu_gpu_fusion_test.cc91 auto hlo_module = CreateNewVerifiedModule(); in TestElementwise2D() local
124 hlo_module->AddEntryComputation(builder.Build()) in TestElementwise2D()
130 auto actual = ExecuteAndTransfer(std::move(hlo_module), {}); in TestElementwise2D()
201 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
233 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
241 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
249 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
261 hlo_module->AddEntryComputation(builder.Build()) in XLA_TEST_F()
267 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
284 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.cc45 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization() argument
51 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment() argument
113 auto hlo_module = CreateNewVerifiedModule(); in TestCompilerHooks() local
114 hlo_module->AddEntryComputation(builder.Build()); in TestCompilerHooks()
120 ->RunBackend(std::move(hlo_module), in TestCompilerHooks()
135 std::unique_ptr<HloModule> hlo_module = CreateNewVerifiedModule(); in TestMultiModuleCompilation() local
136 hlo_module->AddEntryComputation(builder.Build()); in TestMultiModuleCompilation()
139 module_group->push_back(hlo_module->Clone()); in TestMultiModuleCompilation()
140 module_group->push_back(std::move(hlo_module)); in TestMultiModuleCompilation()
/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.cc145 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
148 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
159 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
179 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
182 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
193 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
215 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
218 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F()
229 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
245 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
[all …]
Dxla_debug_info_manager.cc23 const ModuleIdentifier& module_id, std::shared_ptr<HloModule> hlo_module, in RegisterModule() argument
27 active_modules_[module_id].instances.emplace_back(hlo_module, in RegisterModule()
32 m.instances.emplace_back(hlo_module, buffer_assignment); in RegisterModule()
41 const ModuleIdentifier& module_id, std::shared_ptr<HloModule> hlo_module, in UnregisterModule() argument
48 return e.hlo_module == hlo_module && in UnregisterModule()
147 if (m.instances[0].hlo_module && m.instances[0].buffer_assignment) { in StopTracing()
149 MakeHloProto(*m.instances[0].hlo_module)); in StopTracing()
Dhlo_proto_util.cc59 if (!hlo_proto.hlo_module().has_host_program_shape()) { in EntryComputationParameterShapes()
64 const auto& program_shape = hlo_proto.hlo_module().host_program_shape(); in EntryComputationParameterShapes()
76 if (!hlo_proto.hlo_module().has_host_program_shape()) { in EntryComputationOutputShape()
79 if (!hlo_proto.hlo_module().host_program_shape().has_result()) { in EntryComputationOutputShape()
83 return &hlo_proto.hlo_module().host_program_shape().result(); in EntryComputationOutputShape()
Dpattern_matcher_test.cc38 TF_ASSERT_OK_AND_ASSIGN(auto hlo_module, in TEST_F()
47 hlo_module->entry_computation()->root_instruction(), in TEST_F()
61 EXPECT_TRUE(Match(hlo_module->entry_computation()->root_instruction(), in TEST_F()
64 EXPECT_FALSE(Match(hlo_module->entry_computation()->root_instruction(), in TEST_F()
67 EXPECT_FALSE(Match(hlo_module->entry_computation()->root_instruction(), in TEST_F()
165 TF_ASSERT_OK_AND_ASSIGN(auto hlo_module, in TEST_F()
168 auto* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
185 TF_ASSERT_OK_AND_ASSIGN(auto hlo_module, in TEST_F()
188 auto* root = hlo_module->entry_computation()->root_instruction(); in TEST_F()
199 TF_ASSERT_OK_AND_ASSIGN(auto hlo_module, in TEST_F()
[all …]
Dhlo_execution_profile_test.cc31 auto hlo_module = ParseAndReturnVerifiedModule(R"( in TEST_F() local
41 hlo_module->entry_computation()->root_instruction(); in TEST_F()
54 HloProfileIndexMap profile_index_map(*hlo_module); in TEST_F()
57 hlo_module->entry_computation()->name()); in TEST_F()
/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 …]
Dhlo_module_loader_test.cc39 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> hlo_module, in TEST_F()
41 EXPECT_NE(FindInstruction(hlo_module.get(), "p0"), nullptr); in TEST_F()
42 EXPECT_NE(FindInstruction(hlo_module.get(), "p1"), nullptr); in TEST_F()
43 EXPECT_NE(FindInstruction(hlo_module.get(), "add"), nullptr); in TEST_F()
44 EXPECT_NE(FindInstruction(hlo_module.get(), "rooty"), nullptr); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc81 Status InterpreterCompiler::RunHloOptimization(HloModule* hlo_module) { in RunHloOptimization() argument
90 hlo_module->mutable_entry_computation_layout(), in RunHloOptimization()
93 return pipeline.Run(hlo_module).status(); in RunHloOptimization()
97 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses() argument
99 VLOG(1) << "Run hlo passes on graph " << hlo_module->name(); in RunHloPasses()
100 TF_RETURN_IF_ERROR(RunHloOptimization(hlo_module.get())); in RunHloPasses()
101 return std::move(hlo_module); in RunHloPasses()
105 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend() argument
109 VLOG(1) << "Run backend " << hlo_module->name(); in RunBackend()
112 DynamicDimensionInference::Run(hlo_module.get())); in RunBackend()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc150 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloModule() argument
199 if (hlo_module->config().debug_options().xla_gpu_use_cudnn_batchnorm()) { in OptimizeHloModule()
280 TF_RETURN_IF_ERROR(pipeline.Run(hlo_module).status()); in OptimizeHloModule()
286 hlo_module, stream_exec, device_allocator)); in OptimizeHloModule()
300 hlo_module->mutable_entry_computation_layout(), in OptimizeHloModule()
302 TF_RETURN_IF_ERROR(pipeline.Run(hlo_module).status()); in OptimizeHloModule()
306 TF_RETURN_IF_ERROR(OptimizeHloPostLayoutAssignment(hlo_module, stream_exec, in OptimizeHloModule()
327 TF_RETURN_IF_ERROR(fusion.Run(hlo_module).status()); in OptimizeHloModule()
335 TF_RETURN_IF_ERROR(horizontal_fusion.Run(hlo_module).status()); in OptimizeHloModule()
343 TF_RETURN_IF_ERROR(pipeline.Run(hlo_module).status()); in OptimizeHloModule()
[all …]
Dgpu_compiler.h62 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,
94 Status PrepareHloModuleForIrEmitting(HloModule* hlo_module);
154 HloModule* hlo_module, llvm::LLVMContext* llvm_context,
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc91 std::shared_ptr<HloModule> hlo_module) in TpuExecutable() argument
92 : TpuExecutableInterface(std::move(hlo_module)), in TpuExecutable()
222 XLA_HloModule hlo_module; in RunHloPasses() local
223 auto cleanup = xla::MakeCleanup([&hlo_module]() { in RunHloPasses()
224 stream_executor::tpu::SerializedProto_Free(hlo_module.proto); in RunHloPasses()
225 ApiConverter::Free(&hlo_module.module_config); in RunHloPasses()
227 hlo_module.module_config = ApiConverter::ToC(module->config()); in RunHloPasses()
228 hlo_module.proto = stream_executor::tpu::SerializeProto(module->ToProto()); in RunHloPasses()
233 compiler_, &hlo_module, in RunHloPasses()
250 XLA_HloModule hlo_module; in RunBackend() local
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Ddynamic_parameter_binding.mlir10 // CHECK-LABEL: hlo_module
14 // TUPLE-LABEL: hlo_module
26 // CHECK-LABEL: hlo_module
33 // TUPLE-LABEL: hlo_module
49 // CHECK-LABEL: hlo_module
60 // TUPLE-LABEL: hlo_module
82 // CHECK-LABEL: hlo_module
98 // TUPLE-LABEL: hlo_module
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate.cc94 auto hlo_module = std::move(hlo_module_error.ValueOrDie()); in HloTextToMlirHloTranslateFunction() local
98 ConvertHloToMlirHlo(*module, hlo_module.get(), import_all_computations); in HloTextToMlirHloTranslateFunction()
125 const HloModuleProto& module_proto = hlo_proto.hlo_module(); in HloModuleFromProto()
164 auto hlo_module = computation.proto(); in ConvertMlirHloToHloViaBuilder() local
165 hlo_proto->mutable_hlo_module()->Swap(&hlo_module); in ConvertMlirHloToHloViaBuilder()
197 HloModule* hlo_module = statusOrHloModule.ValueOrDie().get(); in MlirHloToHloTextTranslateFunctionImpl() local
199 output << hlo_module->ToString( in MlirHloToHloTextTranslateFunctionImpl()
203 hlo_module->input_output_alias_config().ForEachAlias( in MlirHloToHloTextTranslateFunctionImpl()

1234