HomeSort by relevance Sort by last modified time
    Searched refs:ptx_by_compute_capability_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
kernel_spec.cc 54 ptx_by_compute_capability_(CompareComputeCapability) {
60 ptx_by_compute_capability_[kMinimumCapability] = ptx.data();
67 ptx_by_compute_capability_(CompareComputeCapability) {
77 ptx_by_compute_capability_[std::tuple<int, int>{major, minor}] = ptx.data();
94 if (ptx_by_compute_capability_.empty()) {
100 auto ptx = ptx_by_compute_capability_.begin()->second;
115 if (ptx_by_compute_capability_.empty()) {
119 return ptx_by_compute_capability_.begin()->second;
127 auto ptx_iter = ptx_by_compute_capability_.find(capability);
128 if (ptx_iter == ptx_by_compute_capability_.end())
    [all...]
kernel_spec.h 205 ptx_by_compute_capability_; member in class:perftools::gputools::CudaPtxInMemory

Completed in 7800 milliseconds