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

  /external/tensorflow/tensorflow/core/kernels/
check_numerics_op_gpu.cu.cc 65 (d.getNumCudaMultiProcessors() * d.maxCudaThreadsPerMultiProcessor()) /
random_op_gpu.cu.cc 222 (d.getNumCudaMultiProcessors() * d.maxCudaThreadsPerMultiProcessor()) /
depthwise_conv_op_gpu.cu.cc 705 : device.getNumCudaMultiProcessors();
    [all...]
  /external/tensorflow/tensorflow/core/util/
cuda_launch_config.h 131 d.getNumCudaMultiProcessors() * d.maxCudaThreadsPerMultiProcessor(),
136 d.getNumCudaMultiProcessors());
187 block_count = std::min(block_count * d.getNumCudaMultiProcessors(),
216 d.getNumCudaMultiProcessors() * d.maxCudaThreadsPerMultiProcessor();
  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorReductionCuda.h 562 const int max_blocks = device.getNumCudaMultiProcessors() *
570 const int max_blocks = device.getNumCudaMultiProcessors() *
608 const int max_blocks = device.getNumCudaMultiProcessors() *
616 const int max_blocks = device.getNumCudaMultiProcessors() *
721 const int max_blocks = device.getNumCudaMultiProcessors() *
729 const int max_blocks = device.getNumCudaMultiProcessors() *
TensorDeviceCuda.h 278 EIGEN_STRONG_INLINE int getNumCudaMultiProcessors() const {
TensorExecutor.h 254 const int max_blocks = device.getNumCudaMultiProcessors() *
TensorConvolution.h     [all...]

Completed in 1286 milliseconds