Home
last modified time | relevance | path

Searched defs:CudnnSetTensorNdDescriptor (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/
Dgpu_kernel.cc336 bool CudnnSetTensorNdDescriptor(const ShapeVector &shape, cudnnTensorDescriptor_t descriptor, cudnn… in CudnnSetTensorNdDescriptor() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Deinsum_helper.h304 …void CudnnSetTensorNdDescriptor(const std::vector<size_t> &shape, cudnnTensorDescriptor_t descript… in CudnnSetTensorNdDescriptor() function