Searched refs:GpuAllocator (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 152 test_case->ins.emplace_back(GpuAllocator(device), data_type_, shape); in MakeReductionTestCase()
153 test_case->outs.emplace_back(GpuAllocator(device), data_type_, shape); in MakeReductionTestCase()
192 test_case->ins.emplace_back(GpuAllocator(device), data_type_, in_shape); in MakeGatherTestCase()
193 test_case->outs.emplace_back(GpuAllocator(device), data_type_, in MakeGatherTestCase()
222 test_case->ins.emplace_back(GpuAllocator(device), data_type_, shape); in MakeBroadcastTestCase()
226 test_case->outs.emplace_back(GpuAllocator(device), data_type_, in MakeBroadcastTestCase()
239 test_case->outs.emplace_back(GpuAllocator(device), data_type_, shape); in MakeBroadcastTestCase()
435 static Allocator* GpuAllocator(BaseGPUDevice* device) { in GpuAllocator() function in tensorflow::NcclManagerTest
|