Home
last modified time | relevance | path

Searched defs:src_coordinate (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dmatrix_split_impl.cu26 size_t src_coordinate = (batchIdx * split_dim + dst_x) * dim + batchIdx * split_dim + dst_y; in MatrixSplitKernel() local
38 size_t src_coordinate = (batchIdx * split_dim + dst_x) * dim + batchIdx * split_dim + dst_y; in MatrixSplitKernel() local
Dconvert_gradient_impl.cu28 …size_t src_coordinate = dst_batchIdxX * height_h * width + dst_x * width + dst_batchIdxY * height_… in ConvertGradientKernel() local
42 …size_t src_coordinate = dst_batchIdxX * height_h * width + dst_x * width + dst_batchIdxY * height_… in ConvertGradientBackKernel() local
60 size_t src_coordinate = src_x * ori_w + src_y; in ConvertGradientBackKernel() local
Dmatrix_combine_impl.cu47 size_t src_coordinate = (batch - 1) * src_height * src_width + src_h * src_width + src_w; in MatrixCombineKernel() local