Searched refs:GetConvAccumulatorType (Results 1 – 1 of 1) sorted by relevance
2685 dnn::DataType GetConvAccumulatorType(dnn::DataType data_type);2699 ToCudnnDataType(GetConvAccumulatorType(element_type))); in GetCudnnConvolutionForwardAlgorithm()2761 ToCudnnDataType(GetConvAccumulatorType(element_type))); in GetCudnnConvolutionBackwardDataAlgorithm()2822 ToCudnnDataType(GetConvAccumulatorType(element_type))); in GetCudnnConvolutionBackwardFilterAlgorithm()2965 dnn::DataType GetConvAccumulatorType(dnn::DataType data_type) { in GetConvAccumulatorType() function3053 auto accumulator_type = GetConvAccumulatorType(element_type); in DoConvolve()3774 GetConvAccumulatorType(dnn::DataType::kDouble), scratch_allocator, in DoFusedConvolve()3796 GetConvAccumulatorType(dnn::DataType::kFloat), scratch_allocator, in DoFusedConvolve()3819 GetConvAccumulatorType(dnn::DataType::kHalf), scratch_allocator, in DoFusedConvolve()3850 GetConvAccumulatorType(dnn::DataType::kInt8), scratch_allocator, in DoFusedConvolve()[all …]