Searched refs:output_nd (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2159 const CudnnTensorDescriptor& output_nd, bool specify_workspace_limit, in GetCudnnConvolutionForwardAlgo() argument
2167 output_nd.handle(), preference, memory_limit_bytes, &algo_to_use)); in GetCudnnConvolutionForwardAlgo()
2176 const CudnnTensorDescriptor& output_nd, in GetCudnnConvolutionBackwardDataAlgo() argument
2185 cudnn.handle(), filter.handle(), output_nd.handle(), conv.handle(), in GetCudnnConvolutionBackwardDataAlgo()
2195 const CudnnTensorDescriptor& output_nd, in GetCudnnConvolutionBackwardFilterAlgo() argument
2204 cudnn.handle(), input_nd.handle(), output_nd.handle(), conv.handle(), in GetCudnnConvolutionBackwardFilterAlgo()
2213 const CudnnTensorDescriptor& output_nd, in AllocateCudnnConvolutionForwardWorkspace() argument
2227 /*yDesc=*/output_nd.handle(), /*algo=*/ToConvForwardAlgo(algorithm_desc), in AllocateCudnnConvolutionForwardWorkspace()
2256 const CudnnTensorDescriptor& output_nd, in AllocateCudnnConvolutionBackwardDataWorkspace() argument
2269 /*dyDesc=*/output_nd.handle(), in AllocateCudnnConvolutionBackwardDataWorkspace()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2607 ScopedTensorDescriptor output_nd{ in DoPrepareForConvolution() local
2635 /*destDesc=*/output_nd.handle(), /*sizeInBytes=*/&size_in_bytes); in DoPrepareForConvolution()
2640 miopen.handle(), /*diffDesc=*/output_nd.handle(), in DoPrepareForConvolution()
2647 miopen.handle(), /*diffDesc=*/output_nd.handle(), in DoPrepareForConvolution()
2672 output_nd.handle(), output_data.opaque(), in DoPrepareForConvolution()
2687 /*diffDesc=*/output_nd.handle(), output_data.opaque(), in DoPrepareForConvolution()
2705 /*diffDesc=*/output_nd.handle(), output_data.opaque(), in DoPrepareForConvolution()
2821 ScopedTensorDescriptor output_nd{output_descriptor, in DoConvolve() local
2860 /*beta=*/&beta, /*destDesc=*/output_nd.handle(), in DoConvolve()
2878 /*diffDesc=*/output_nd.handle(), in DoConvolve()
[all …]
|