Searched refs:filter_nd (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2996 CudnnFilterDescriptor filter_nd( in DoPrepareForConvolution() local
3010 filter_nd, element_type, convolution_descriptor, in DoPrepareForConvolution()
3018 filter_nd, element_type, convolution_descriptor, in DoPrepareForConvolution()
3026 filter_nd, element_type, convolution_descriptor, in DoPrepareForConvolution()
3052 CudnnFilterDescriptor filter_nd(filter_descriptor, cudnn_type); in DoConvolve() local
3113 /*srcData=*/input_data.opaque(), /*filterDesc=*/filter_nd.handle(), in DoConvolve()
3126 /*wDesc=*/filter_nd.handle(), in DoConvolve()
3153 /*gradDesc=*/filter_nd.handle(), in DoConvolve()
|