nccl_manager_test.cc | 64 static Allocator* gpu_allocator(BaseGPUDevice* device) { function in class:tensorflow::NcclManagerTest 140 test_case->ins.emplace_back(gpu_allocator(device), DT_FLOAT, shape); 141 test_case->outs.emplace_back(gpu_allocator(device), DT_FLOAT, shape);
|