OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Cleanup code

Browse Source
This commit is contained in:
Kaixi Hou
2021-03-05 15:55:11 -08:00
parent a047609b55
commit d5470a4018
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/stream_executor/cuda/cuda_dnn.cc
View File

@@ -3483,7 +3483,7 @@ port::Status CudnnSupport::DoPrepareForConvolution(
return port::Status::OK();
}
port::Status CudnnSupport::DoConvolveWithExecutionPlan(
port::Status CudnnSupport::DoConvolve(
dnn::ConvolutionKind kind, dnn::DataType element_type,
dnn::DataType output_type, Stream* stream,
const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
@@ -3620,7 +3620,7 @@ port::Status CudnnSupport::DoConvolveWithExecutionPlan(
return port::Status::OK();
}
port::Status CudnnSupport::DoConvolve(
port::Status CudnnSupport::DoConvolveWithExecutionPlan(
dnn::ConvolutionKind kind, dnn::DataType element_type,
dnn::DataType output_type, Stream* stream,
const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,