Searched defs:ptx_iter (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
H A Dkernel_spec.cc127 auto ptx_iter = ptx_by_compute_capability_.find(capability); local
128 if (ptx_iter == ptx_by_compute_capability_.end()) {
135 auto decompressed_ptx_iter = decompressed_ptx_.find(ptx_iter->second);
140 decompressed_ptx_iter->second = DecompressPtx(ptx_iter->second);
144 return ptx_iter->second;
152 auto ptx_iter = ptx_by_compute_capability_.find(capability); local
153 if (ptx_iter == ptx_by_compute_capability_.end()) {
157 return ptx_iter->second;

Completed in 126 milliseconds