Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_kernel_tiling_test.cc63 auto hlo_module = in TEST_F() local
89 auto hlo_module = in TEST_F() local
116 auto hlo_module = in TEST_F() local
150 auto hlo_module = in TEST_F() local
185 auto hlo_module = in TEST_F() local
212 auto hlo_module = in TEST_F() local
239 auto hlo_module = in TEST_F() local
274 auto hlo_module = in TEST_F() local
305 auto hlo_module = in TEST_F() local
351 auto hlo_module = in TEST_F() local
[all …]
Dgpu_unrolling_test.cc51 auto hlo_module = ParseHloString(kAddModule, config).ValueOrDie(); in TEST_F() local
70 auto hlo_module = ParseHloString(kAddModule, config).ValueOrDie(); in TEST_F() local
89 auto hlo_module = ParseHloString(kAddModule, config).ValueOrDie(); in TEST_F() local
120 auto hlo_module = ParseHloString(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() local
163 auto hlo_module = in TEST_F() local
Dgpu_ldg_test.cc51 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
76 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
98 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_ftz_test.cc42 auto hlo_module = CreateNewUnverifiedModuleWithFTZ(ftz_); in CreateBinaryOpModule() local
57 auto hlo_module = CreateNewUnverifiedModuleWithFTZ(ftz_); in CreateUnaryOpModule() local
Dgpu_copy_test.cc48 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_codegen_test.cc39 void GpuCodegenTest::CompileAndVerifyPtx(std::unique_ptr<HloModule> hlo_module, in CompileAndVerifyPtx()
Dgpu_noalias_test.cc50 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_index_test.cc55 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc45 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
61 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
84 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
105 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
124 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
141 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
161 auto hlo_module = CreateNewUnverifiedModule(); in TEST_F() local
186 auto hlo_module = CreateNewUnverifiedModule(); in TEST_F() local
217 auto hlo_module = CreateNewUnverifiedModule(); in XLA_TEST_F() local
233 auto hlo_module = CreateNewUnverifiedModule(); in TEST_F() local
[all …]
Dfusion_test.cc92 auto hlo_module = CreateNewVerifiedModule(); in TestElementwise2D() local
202 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
250 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
285 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
309 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
333 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
348 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
363 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
378 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
393 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_irgen_test_base.cc51 std::unique_ptr<HloModule> hlo_module, const string& pattern, in CompileAndVerifyIr()
74 std::unique_ptr<HloModule> hlo_module, const AotCompilationOptions& options, in CompileAheadOfTimeAndVerifyIr()
Dllvm_compiler_test.cc71 auto hlo_module = CreateNewUnverifiedModule(); in TestCompilerHooks() local
93 std::unique_ptr<HloModule> hlo_module = CreateNewUnverifiedModule(); in TestMultiModuleCompilation() local
Dmultioutput_fusion_test.cc65 auto hlo_module = CreateNewUnverifiedModule(); in RunTest2D() local
125 auto hlo_module = CreateNewUnverifiedModule(); in RunTest1D() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc146 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
175 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
204 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
227 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
248 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
274 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
303 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
324 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
342 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
361 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.cc32 auto hlo_module = ParseHloString(R"( in TEST_F() local
Dexecutable.h64 std::unique_ptr<HloModule> hlo_module, in Executable()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h33 IrEmitterContext(const HloModule* hlo_module, in IrEmitterContext()
46 const HloModule& hlo_module() const { return *hlo_module_; } in hlo_module() function
Dnvptx_compiler.cc179 Status OptimizeHloModule(HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloModule()
398 Status PrepareHloModuleForIrEmitting(HloModule* hlo_module) { in PrepareHloModuleForIrEmitting()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc80 Status InterpreterCompiler::RunHloOptimization(HloModule* hlo_module) { in RunHloOptimization()
98 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
113 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend()
Dexecutable.cc40 std::unique_ptr<HloModule> hlo_module, in InterpreterExecutable()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_eigen_dot_operation_test.cc52 auto hlo_module = CreateNewVerifiedModule(); in CompileAndCheck() local
Dcpu_noalias_test.cc59 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dcpu_intrinsic_test.cc95 auto hlo_module = CreateNewVerifiedModule(); in TEST_P() local

12