Searched refs:GetCudnnDataType (Results 1 – 1 of 1) sorted by relevance
874 cudnnDataType_t GetCudnnDataType( in GetCudnnDataType() function2945 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl()2948 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl()2951 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl()2952 CudnnTensorDescriptor bias_nd(bias_descriptor, GetCudnnDataType<BiasType>()); in DoFusedConvolveImpl()3476 cudnnDataType_t cudnn_type = GetCudnnDataType<T>(); in DoConvolveBackwardBiasImpl()