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

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorDeviceCuda.h 15 static const int kCudaScratchSize = 1024;
160 scratch_ = allocate(kCudaScratchSize + sizeof(unsigned int));
167 char* scratch = static_cast<char*>(scratchpad()) + kCudaScratchSize;
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc 104 reinterpret_cast<unsigned int*>(scratch + Eigen::kCudaScratchSize);
290 size_t scratch_buffer_size = Eigen::kCudaScratchSize + sizeof(unsigned int);
304 &mem, Eigen::kCudaScratchSize + sizeof(unsigned int));
    [all...]

Completed in 90 milliseconds