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

  /external/tensorflow/tensorflow/stream_executor/
kernel.cc 46 bool KernelMetadata::shared_memory_bytes(int *shared_memory_bytes) const { function in class:perftools::gputools::KernelMetadata
48 *shared_memory_bytes = shared_memory_bytes_;
55 void KernelMetadata::set_shared_memory_bytes(int shared_memory_bytes) {
56 shared_memory_bytes_ = shared_memory_bytes;
kernel.h 116 bool shared_memory_bytes(int *shared_memory_bytes) const;
120 void set_shared_memory_bytes(int shared_memory_bytes);
127 // Holds the value returned by shared_memory_bytes above.
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc 410 if (!kernel.metadata().shared_memory_bytes(&smem_per_block)) {
    [all...]

Completed in 213 milliseconds