p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

batchtospace

This commit is contained in:
zong_shuai 2021-08-24 14:55:10 +08:00
parent f518b9e9fd
commit 48732fb104
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/batchtospace_gpu_kernel.h
View File

@ -29,7 +29,7 @@ template <typename T>
class BatchToSpaceGpuKernel : public GpuKernel {
public:
BatchToSpaceGpuKernel() { ResetResource(); }
~BatchToSpaceGpuKernel() {}
~BatchToSpaceGpuKernel() = default;
const std::vector<size_t> &GetInputSizeList() const override { return input_size_list_; }
const std::vector<size_t> &GetOutputSizeList() const override { return output_size_list_; }
const std::vector<size_t> &GetWorkspaceSizeList() const override { return workspace_size_list_; }
@ -53,6 +53,7 @@ class BatchToSpaceGpuKernel : public GpuKernel {
}
input_size_ = sizeof(T);
for(size_t idx = 0; idx < shape_size_; ++idx){
printf("idx:%zd,input_size:%zd,input_shape:%zd\n",idx,input_size_,input_shape_[idx]);
input_size_ *= input_shape_[idx];
}