Home
last modified time | relevance | path

Searched refs:GetOptimizedModule (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dllvm_irgen_test_base.cc90 GetOptimizedModule(hlo)); in MatchOptimizedHlo()
99 StatusOr<std::unique_ptr<HloModule>> LlvmIrGenTestBase::GetOptimizedModule( in GetOptimizedModule() function in xla::LlvmIrGenTestBase
109 StatusOr<std::unique_ptr<HloModule>> LlvmIrGenTestBase::GetOptimizedModule( in GetOptimizedModule() function in xla::LlvmIrGenTestBase
H A Dllvm_irgen_test_base.h73 StatusOr<std::unique_ptr<HloModule>> GetOptimizedModule(
76 StatusOr<std::unique_ptr<HloModule>> GetOptimizedModule(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dgpu_too_many_blocks_test.cc46 GetOptimizedModule(hlo_text)); in TEST_F()
H A Dgpu_spmd_e2e_compile_test.cc75 GetOptimizedModule(std::move(hlo_module))); in TEST_F()
H A Dgemm_rewrite_test.cc48 GetOptimizedModule(hlo)); in CheckNumberOfAllocations()