Searched refs:GPUDeviceTestHelper (Results 1 – 2 of 2) sorted by relevance
144 class GPUDeviceTestHelper { class146 GPUDeviceTestHelper(size_t memory_limit, int pending_cap) { in GPUDeviceTestHelper() function in tensorflow::GPUDeviceTestHelper177 GPUDeviceTestHelper* gpu_helper_;204 EMBenchmarkHelper(GPUDeviceTestHelper* h) : gpu_helper_(h) {} in EMBenchmarkHelper()474 GPUDeviceTestHelper* gpu_helper = nullptr;500 gpu_helper = new GPUDeviceTestHelper(1 << 24, pending_cap); in BM_chain_ops()
130 friend class GPUDeviceTestHelper;