Searched refs:LocalClientTestBase (Results 1 – 8 of 8) sorted by relevance
42 /* static */ TestAllocator* LocalClientTestBase::allocator_;98 /* static */ TestAllocator* LocalClientTestBase::GetOrCreateAllocator( in GetOrCreateAllocator()113 struct LocalClientTestBase::EigenThreadPoolWrapper {124 LocalClientTestBase::LocalClientTestBase(se::Platform* platform) in LocalClientTestBase() function in xla::LocalClientTestBase136 LocalClientTestBase::~LocalClientTestBase() {} in ~LocalClientTestBase()138 ScopedShapedBuffer LocalClientTestBase::LiteralToShapedBuffer( in LiteralToShapedBuffer()145 Literal LocalClientTestBase::ShapedBufferToLiteral( in ShapedBufferToLiteral()151 ExecutableBuildOptions LocalClientTestBase::DefaultExecutableBuildOptions() in DefaultExecutableBuildOptions()156 ExecutableRunOptions LocalClientTestBase::DefaultExecutableRunOptions() const { in DefaultExecutableRunOptions()163 ScopedShapedBuffer LocalClientTestBase::ExecuteLocallyOrDie( in ExecuteLocallyOrDie()[all …]
79 class LocalClientTestBase : public ManifestCheckingTest {82 explicit LocalClientTestBase(se::Platform* platform = nullptr);83 virtual ~LocalClientTestBase();
27 class HloMetadataTest : public LocalClientTestBase {
34 class LocalClientAllocationTest : public LocalClientTestBase {
26 class OutfeedInNestedComputationTest : public LocalClientTestBase {};
31 class TestUtilsTest : public LocalClientTestBase {};
41 class TransferManagerTest : public LocalClientTestBase {
51 class LocalClientExecuteTest : public LocalClientTestBase {