Searched refs:filter_nd (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2638 CudnnFilterDescriptor filter_nd( in DoPrepareForConvolution() local
2655 stream, cudnn, algorithm_config, input_nd, filter_nd, conv, in DoPrepareForConvolution()
2663 stream, cudnn, algorithm_config, input_nd, filter_nd, conv, in DoPrepareForConvolution()
2671 stream, cudnn, algorithm_config, input_nd, filter_nd, conv, in DoPrepareForConvolution()
2695 CudnnFilterDescriptor filter_nd(filter_descriptor, cudnn_type); in DoConvolve() local
2859 /*srcData=*/input_data.opaque(), /*filterDesc=*/filter_nd.handle(), in DoConvolve()
2872 /*wDesc=*/filter_nd.handle(), in DoConvolve()
2899 /*gradDesc=*/filter_nd.handle(), in DoConvolve()
|