Searched refs:cudnn_dtype (Results 1 – 1 of 1) sorted by relevance
| /dports/science/py-chainer/chainer-7.8.0/chainerx_cc/chainerx/cuda/ |
| H A D | cudnn.cc | 100 cudnnDataType_t cudnn_dtype = GetCudnnDataType(arr.dtype()); local
103 …CheckCudnnError(cudnnSetTensor4dDescriptor(desc_, CUDNN_TENSOR_NCHW, cudnn_dtype, nchw[0], nchw[1]…
107 …CheckCudnnError(cudnnSetTensorNdDescriptor(desc_, cudnn_dtype, arr.ndim(), &int_shape[0], &int_str…
112 cudnnDataType_t cudnn_dtype{}; in GetDtype() local
115 CheckCudnnError(cudnnGetTensorNdDescriptor(desc_, 0, &cudnn_dtype, &ndim, nullptr, nullptr)); in GetDtype()
117 switch (cudnn_dtype) { in GetDtype()
125 throw DtypeError{"Unsupported cudnn data type: ", cudnn_dtype}; in GetDtype()
140 cudnnDataType_t cudnn_dtype = GetCudnnDataType(w.dtype()); local
143 …CheckCudnnError(cudnnSetFilter4dDescriptor(desc_, cudnn_dtype, CUDNN_TENSOR_NCHW, nchw[0], nchw[1]…
146 …CheckCudnnError(cudnnSetFilterNdDescriptor(desc_, cudnn_dtype, CUDNN_TENSOR_NCHW, w.ndim(), &int_s…
|