Home
last modified time | relevance | path

Searched defs:hlo_module (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_kernel_tiling_test.cc63 auto hlo_module = in TEST_F() local
97 auto hlo_module = in TEST_F() local
146 auto hlo_module = in TEST_F() local
186 auto hlo_module = in TEST_F() local
227 auto hlo_module = in TEST_F() local
260 auto hlo_module = in TEST_F() local
293 auto hlo_module = in TEST_F() local
334 auto hlo_module = in TEST_F() local
371 auto hlo_module = in TEST_F() local
435 auto hlo_module = in TEST_F() local
[all …]
Dgpu_unrolling_test.cc51 auto hlo_module = in TEST_F() local
71 auto hlo_module = in TEST_F() local
91 auto hlo_module = in TEST_F() local
123 auto hlo_module = in TEST_F() local
154 auto hlo_module = in TEST_F() local
188 auto hlo_module = in TEST_F() local
222 auto hlo_module = in TEST_F() local
247 auto hlo_module = in TEST_F() local
285 auto hlo_module = in TEST_F() local
Dgpu_input_fusible_slice_test.cc63 auto hlo_module = in TEST_F() local
105 auto hlo_module = in TEST_F() local
152 auto hlo_module = in TEST_F() local
Dgpu_ldg_test.cc56 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
86 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
113 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_ftz_test.cc45 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateBinaryOpModule() local
60 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateUnaryOpModule() local
Dgpu_copy_test.cc48 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_codegen_test.cc50 std::unique_ptr<VerifiedHloModule> hlo_module, absl::string_view pattern) { in CompileAndOptionallyVerifyPtx()
Dgpu_noalias_test.cc50 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc45 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
61 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
84 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
105 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
124 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
141 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
161 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
186 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
217 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
233 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
[all …]
Dcpu_gpu_fusion_test.cc91 auto hlo_module = CreateNewVerifiedModule(); in TestElementwise2D() local
201 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
249 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
284 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
308 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
332 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
347 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
362 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
377 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
392 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
[all …]
Dcodegen_test_base.cc21 std::unique_ptr<HloModule> hlo_module) { in CompileToExecutable()
33 std::unique_ptr<HloModule> hlo_module, in CompileToAotCompilationResult()
Dllvm_compiler_test.cc45 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
51 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
59 const HloModule* hlo_module, llvm::Module* llvm_module, in CompileTargetBinary()
114 auto hlo_module = CreateNewVerifiedModule(); in TestCompilerHooks() local
136 std::unique_ptr<HloModule> hlo_module = CreateNewVerifiedModule(); in TestMultiModuleCompilation() local
Dllvm_irgen_test_base.cc50 std::unique_ptr<HloModule> hlo_module, const string& pattern, in CompileAndVerifyIr()
73 std::unique_ptr<HloModule> hlo_module, const AotCompilationOptions& options, in CompileAheadOfTimeAndVerifyIr()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc145 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
179 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
215 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
245 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
272 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
302 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
336 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
359 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
382 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
411 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
[all …]
Dmap_inliner_test.cc62 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
96 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
134 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
166 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dhlo_execution_profile_test.cc31 auto hlo_module = ParseAndReturnVerifiedModule(R"( in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dmlir_irgen_test_base.cc40 void MlirIrGenTestBase::CompileIr(std::unique_ptr<HloModule> hlo_module, in CompileIr()
56 std::unique_ptr<HloModule> hlo_module, in CompileIr()
73 std::unique_ptr<HloModule> hlo_module, const string& pattern, in CompileAndVerifyIr()
106 std::unique_ptr<HloModule> hlo_module, LoweringStage breaking_stage) { in CompileAndInjectErrors()
109 HloModule* hlo_module) { in CompileAndInjectErrors()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc79 Status InterpreterCompiler::RunHloOptimization(HloModule* hlo_module) { in RunHloOptimization()
93 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
101 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h33 IrEmitterContext(const HloModule* hlo_module, in IrEmitterContext()
48 const HloModule& hlo_module() const { return *hlo_module_; } in hlo_module() function
Damdgpu_compiler.cc70 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
87 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
Dgpu_debug_info_manager.cc24 const ModuleIdentifier& module_id, std::shared_ptr<HloModule> hlo_module, in RegisterModule()
42 const ModuleIdentifier& module_id, std::shared_ptr<HloModule> hlo_module, in UnregisterModule()
Dgpu_compiler.cc120 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloModule()
288 Status GpuCompiler::PrepareHloModuleForIrEmitting(HloModule* hlo_module) { in PrepareHloModuleForIrEmitting()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_to_mlir_hlo.cc30 Status ConvertHloToMlirHlo(mlir::ModuleOp module, xla::HloModule* hlo_module) { in ConvertHloToMlirHlo()
Dxla_mlir_translate.cc99 auto hlo_module = std::move(hlo_module_error.ValueOrDie()); in HloTextToMlirHloTranslateFunction() local
156 HloModule* hlo_module = statusOrHloModule.ValueOrDie().get(); in MlirHloToHloTextTranslateFunction() local
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/
Dconv_emitter_test.cc40 VerifiedHloModule hlo_module( in CompileHloConvAndGetMlir() local

12