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

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorDeviceCuda.h 30 // Return a scratchpad buffer of size 1k
31 virtual void* scratchpad() const = 0;
158 virtual void* scratchpad() const {
167 char* scratch = static_cast<char*>(scratchpad()) + kCudaScratchSize;
205 EIGEN_STRONG_INLINE void* scratchpad() const { function
206 return stream_->scratchpad();
TensorReductionCuda.h 321 half2* scratch = static_cast<half2*>(device.scratchpad());

Completed in 579 milliseconds