mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix-Conv3dTranspose-Resize

This commit is contained in:
hbhu_bin 2022-12-06 16:11:54 +08:00
parent 7f6424cf81
commit 78b903f090
3 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/plugin/device/gpu/kernel/nn/conv3d_gpu_kernel.cc
View File

@ -188,6 +188,8 @@ void Conv3dGpuKernelMod::SelectAlgorithm(cudnnTensorDescriptor_t input_descripto
}
void Conv3dGpuKernelMod::SetStrideAndDilation(std::vector<int64_t> stride_me, std::vector<int64_t> dilation_me) {
stride_.clear();
dilation_.clear();
(void)std::transform(stride_me.begin(), stride_me.end(), std::back_inserter(stride_),
[](const int64_t &value) { return static_cast<int>(value); });
(void)std::transform(dilation_me.begin(), dilation_me.end(), std::back_inserter(dilation_),

2
mindspore/ccsrc/plugin/device/gpu/kernel/nn/conv3d_grad_filter_gpu_kernel.h
View File

@ -324,6 +324,8 @@ class Conv3dGradFilterGpuKernelMod : public NativeGpuKernelMod {
}
void SetStrideAndDilation(std::vector<int64_t> stride_me, std::vector<int64_t> dilation_me) {
stride_.clear();
dilation_.clear();
(void)std::transform(stride_me.begin(), stride_me.end(), std::back_inserter(stride_),
[](const int64_t &value) { return static_cast<int>(value); });
(void)std::transform(dilation_me.begin(), dilation_me.end(), std::back_inserter(dilation_),

2
mindspore/ccsrc/plugin/device/gpu/kernel/nn/conv3d_transpose_gpu_kernel.cc
View File

@ -224,6 +224,8 @@ void Conv3dTransposeFwdGpuKernelMod::Set5DDesc(const ShapeVector &input_shape, c
void Conv3dTransposeFwdGpuKernelMod::SetStrideAndDilation(std::vector<int64_t> stride_me,
std::vector<int64_t> dilation_me) {
stride_.clear();
dilation_.clear();
(void)std::transform(stride_me.begin(), stride_me.end(), std::back_inserter(stride_),
[](const int64_t &value) { return static_cast<int>(value); });
(void)std::transform(dilation_me.begin(), dilation_me.end(), std::back_inserter(dilation_),