MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / CreateTensorDescriptor

Function CreateTensorDescriptor

tensorflow/stream_executor/cuda/cuda_dnn.cc:494–498  ·  view source on GitHub ↗

Factory methods for cuDNN types.

Source from the content-addressed store, hash-verified

492
493// Factory methods for cuDNN types.
494TensorDescriptor CreateTensorDescriptor() {
495 cudnnTensorDescriptor_t result;
496 CHECK_CUDNN_OK(cudnnCreateTensorDescriptor(&result));
497 return TensorDescriptor(result);
498}
499#if CUDNN_VERSION >= 7201
500RNNDataDescriptor CreateRNNDataDescriptor() {
501 cudnnRNNDataDescriptor_t result;

Callers 4

CudnnTensorDescriptorMethod · 0.85
cuda_dnn.ccFile · 0.85
CreateMethod · 0.85

Calls 1

TensorDescriptorClass · 0.85

Tested by

no test coverage detected