Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dstrided_slice_grad_gpu_kernel.cc380 void StridedSliceGradGpuKernelMod::ComputeEndMask(std::vector<int64_t> *end, const std::vector<int6… in ComputeEndMask() function in mindspore::kernel::StridedSliceGradGpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dstrided_slice_info.cc95 void StridedSliceInfo::ComputeEndMask() { in ComputeEndMask() function in mindspore::parallel::StridedSliceInfo
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dops_utils.cc98 void ComputeEndMask(std::vector<int64_t> *end, const std::vector<int64_t> &stride, const ShapeVecto… in ComputeEndMask() function