Searched refs:cudnn_type (Results 1 – 1 of 1) sorted by relevance
3048 cudnnDataType_t cudnn_type = ToCudnnDataType(element_type); in DoConvolve() local3049 CudnnTensorDescriptor input_nd(input_descriptor, cudnn_type); in DoConvolve()3052 CudnnFilterDescriptor filter_nd(filter_descriptor, cudnn_type); in DoConvolve()3064 void* alpha = cudnn_type == CUDNN_DATA_DOUBLE ? static_cast<void*>(&dalpha) in DoConvolve()3069 void* beta = cudnn_type == CUDNN_DATA_DOUBLE ? static_cast<void*>(&dbeta) in DoConvolve()4009 cudnnDataType_t cudnn_type = GetCudnnDataType<T>(); in DoConvolveBackwardBiasImpl() local4010 CudnnTensorDescriptor input_nd(input_descriptor, cudnn_type); in DoConvolveBackwardBiasImpl()4011 CudnnTensorDescriptor bias_nd(bias_descriptor, cudnn_type); in DoConvolveBackwardBiasImpl()