D | cuda_dnn.cc | 3974 ScopedTensorDescriptor dest_desc{parent_, output_dimensions, in DoPoolForward() local 3979 src_desc.handle(), input_data.opaque(), &beta, dest_desc.handle(), in DoPoolForward() 4009 ScopedTensorDescriptor dest_desc{parent_, output_dimensions, in DoPoolForward() local 4014 src_desc.handle(), input_data.opaque(), &beta, dest_desc.handle(), in DoPoolForward() 4044 ScopedTensorDescriptor dest_desc{parent_, output_dimensions, CUDNN_DATA_HALF}; in DoPoolForward() local 4048 src_desc.handle(), input_data.opaque(), &beta, dest_desc.handle(), in DoPoolForward() 4080 ScopedTensorDescriptor dest_desc{parent_, output_dimensions, in DoPoolBackward() local 4085 dest_desc.handle(), output_data.opaque(), dest_desc.handle(), in DoPoolBackward() 4118 ScopedTensorDescriptor dest_desc{parent_, output_dimensions, in DoPoolBackward() local 4123 dest_desc.handle(), output_data.opaque(), dest_desc.handle(), in DoPoolBackward() [all …]
|