Searched refs:LocalClientTestBase (Results 1 – 8 of 8) sorted by relevance
37 /* static */ TestAllocator* LocalClientTestBase::allocator_;92 /* static */ TestAllocator* LocalClientTestBase::GetOrCreateAllocator( in GetOrCreateAllocator()107 struct LocalClientTestBase::EigenThreadPoolWrapper {120 LocalClientTestBase::LocalClientTestBase(se::Platform* platform) in LocalClientTestBase() function in xla::LocalClientTestBase130 LocalClientTestBase::~LocalClientTestBase() {} in ~LocalClientTestBase()132 ScopedShapedBuffer LocalClientTestBase::LiteralToShapedBuffer( in LiteralToShapedBuffer()139 Literal LocalClientTestBase::ShapedBufferToLiteral( in ShapedBufferToLiteral()145 ExecutableBuildOptions LocalClientTestBase::DefaultExecutableBuildOptions() in DefaultExecutableBuildOptions()150 ExecutableRunOptions LocalClientTestBase::DefaultExecutableRunOptions() const { in DefaultExecutableRunOptions()157 ScopedShapedBuffer LocalClientTestBase::ExecuteLocallyOrDie( in ExecuteLocallyOrDie()[all …]
74 class LocalClientTestBase : public ::testing::Test {77 explicit LocalClientTestBase(se::Platform* platform = nullptr);78 virtual ~LocalClientTestBase();
25 class HloMetadataTest : public LocalClientTestBase {
34 class LocalClientAllocationTest : public LocalClientTestBase {
26 class OutfeedInNestedComputationTest : public LocalClientTestBase {};
31 class TestUtilsTest : public LocalClientTestBase {};
41 class TransferManagerTest : public LocalClientTestBase {
50 class LocalClientExecuteTest : public LocalClientTestBase {