Searched refs:ToCudnnDataType (Results 1 – 1 of 1) sorted by relevance
903 cudnnDataType_t ToCudnnDataType( in ToCudnnDataType() function923 cudnnDataType_t ToCudnnDataType(dnn::DataType data_type, in ToCudnnDataType() function929 return ToCudnnDataType(data_type); in ToCudnnDataType()935 return ToCudnnDataType(dnn::ToDataType<T>::value, data_layout); in GetCudnnDataType()2086 ToCudnnDataType(data_type), GetRnnComputeType(data_type), in createRnnDescriptor()2099 ToCudnnDataType(data_type))); in createRnnSequenceTensorDescriptor()2112 seq_lengths, time_major, ToCudnnDataType(data_type))); in createRnnSequenceTensorDescriptor()2123 data_size, ToCudnnDataType(data_type))); in createRnnStateTensorDescriptor()2699 ToCudnnDataType(GetConvAccumulatorType(element_type))); in GetCudnnConvolutionForwardAlgorithm()2761 ToCudnnDataType(GetConvAccumulatorType(element_type))); in GetCudnnConvolutionBackwardDataAlgorithm()[all …]