Searched refs:GetCudnnDataType (Results 1 – 1 of 1) sorted by relevance
933 cudnnDataType_t GetCudnnDataType( in GetCudnnDataType() function3200 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl()3203 GetCudnnDataType<OutputType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl()3206 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl()3207 CudnnTensorDescriptor bias_nd(bias_descriptor, GetCudnnDataType<BiasType>()); in DoFusedConvolveImpl()4009 cudnnDataType_t cudnn_type = GetCudnnDataType<T>(); in DoConvolveBackwardBiasImpl()