HomeSort by relevance Sort by last modified time
    Searched defs:TestAllocator (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
local_client_test_base.h 42 class TestAllocator : public StreamExecutorMemoryAllocator {
44 explicit TestAllocator(se::Platform* platform)
80 static TestAllocator* GetOrCreateAllocator(se::Platform* platform);
124 static TestAllocator* allocator_;
  /external/tensorflow/tensorflow/compiler/xla/service/
shaped_buffer_test.cc 46 class TestAllocator : public DeviceMemoryAllocator {
48 TestAllocator()
52 ~TestAllocator() override {
96 TestAllocator allocator;
109 // TestAllocator's destructor checks that all memory was freed.
113 TestAllocator allocator;
155 TestAllocator allocator;

Completed in 3011 milliseconds