Searched refs:ToCudnnDataType (Results 1 – 1 of 1) sorted by relevance
844 cudnnDataType_t ToCudnnDataType( in ToCudnnDataType() function864 cudnnDataType_t ToCudnnDataType(dnn::DataType data_type, in ToCudnnDataType() function870 return ToCudnnDataType(data_type); in ToCudnnDataType()876 return ToCudnnDataType(dnn::ToDataType<T>::value, data_layout); in GetCudnnDataType()1831 ToCudnnDataType(data_type), GetRnnComputeType(data_type), in createRnnDescriptor()1844 ToCudnnDataType(data_type))); in createRnnSequenceTensorDescriptor()1857 seq_lengths, time_major, ToCudnnDataType(data_type))); in createRnnSequenceTensorDescriptor()1868 data_size, ToCudnnDataType(data_type))); in createRnnStateTensorDescriptor()2637 ToCudnnDataType(element_type, input_descriptor.layout())); in DoPrepareForConvolution()2640 ToCudnnDataType(element_type, filter_descriptor.layout())); in DoPrepareForConvolution()[all …]