Searched defs:bias_nd (Results 1 – 2 of 2) sorted by relevance
2952 CudnnTensorDescriptor bias_nd(bias_descriptor, GetCudnnDataType<BiasType>()); in DoFusedConvolveImpl() local3478 CudnnTensorDescriptor bias_nd(bias_descriptor, cudnn_type); in DoConvolveBackwardBiasImpl() local
3237 ScopedTensorDescriptor bias_nd{bias_descriptor, in DoConvolveBackwardBiasImpl() local4037 ScopedTensorDescriptor bias_nd{bias_descriptor, in DoFusedConvolutionBiasActivationImpl() local