Searched refs:pending_cap (Results 1 – 1 of 1) sorted by relevance
146 GPUDeviceTestHelper(size_t memory_limit, int pending_cap) { in GPUDeviceTestHelper() argument164 int pending_cap() { return gpu_->pending_cap_; } in pending_cap() function in tensorflow::GPUDeviceTestHelper480 bool event_after_add, int pending_cap) { in BM_chain_ops() argument483 bool event_after_add, int pending_cap, int threads) { in BM_chain_ops()493 if (gpu_helper && gpu_helper->pending_cap() != pending_cap) { in BM_chain_ops()500 gpu_helper = new GPUDeviceTestHelper(1 << 24, pending_cap); in BM_chain_ops()