Searched refs:input_nd (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2157 const CudnnHandle& cudnn, const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionForwardAlgo() argument
2166 cudnn.handle(), input_nd.handle(), filter.handle(), conv.handle(), in GetCudnnConvolutionForwardAlgo()
2173 const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionBackwardDataAlgo() argument
2186 input_nd.handle(), preference, memory_limit_bytes, &algo_to_use)); in GetCudnnConvolutionBackwardDataAlgo()
2192 const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionBackwardFilterAlgo() argument
2204 cudnn.handle(), input_nd.handle(), output_nd.handle(), conv.handle(), in GetCudnnConvolutionBackwardFilterAlgo()
2211 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionForwardWorkspace() argument
2225 /*xDesc=*/input_nd.handle(), in AllocateCudnnConvolutionForwardWorkspace()
2254 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionBackwardDataWorkspace() argument
2271 /*dxDesc=*/input_nd.handle(), in AllocateCudnnConvolutionBackwardDataWorkspace()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2601 ScopedTensorDescriptor input_nd{ in DoPrepareForConvolution() local
2634 /*srcDesc=*/input_nd.handle(), /*convDesc=*/conv.handle(), in DoPrepareForConvolution()
2642 /*gradDesc=*/input_nd.handle(), /*sizeInBytes=*/&size_in_bytes); in DoPrepareForConvolution()
2648 /*srcDesc=*/input_nd.handle(), /*convDesc=*/conv.handle(), in DoPrepareForConvolution()
2670 miopen.handle(), input_nd.handle(), input_data.opaque(), in DoPrepareForConvolution()
2690 /*gradDesc=*/input_nd.handle(), input_data.opaque(), in DoPrepareForConvolution()
2706 /*srcDesc=*/input_nd.handle(), input_data.opaque(), in DoPrepareForConvolution()
2819 ScopedTensorDescriptor input_nd{input_descriptor, in DoConvolve() local
2855 /*alpha=*/&alpha, /*srcDesc=*/input_nd.handle(), in DoConvolve()
2886 /*gradDesc=*/input_nd.handle(), in DoConvolve()
[all …]
|