Searched defs:batchwidth (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | convert_gradient_impl.cu | 21 … const size_t batchwidth, const size_t width, T *input_addr, T *output_addr) { in ConvertGradientKernel()
35 … const size_t batchwidth, const size_t width, T *input_addr, T *output_addr) { in ConvertGradientBackKernel()
49 … const size_t ori_h, const size_t ori_w, const size_t batchwidth, in ConvertGradientBackKernel()
67 …tGradient(const size_t size, const size_t height_h, const size_t height_w, const size_t batchwidth, in ConvertGradient()
76 … const size_t batchwidth, const size_t width, T *input_addr, T *output_addr, in ConvertGradientBack()
85 … const size_t ori_w, const size_t batchwidth, const size_t width, T *input_addr, in ConvertGradientBack()
|