Searched defs:CudnnSetTensorNdDescriptor (Results 1 – 2 of 2) sorted by relevance
336 bool CudnnSetTensorNdDescriptor(const ShapeVector &shape, cudnnTensorDescriptor_t descriptor, cudnn… in CudnnSetTensorNdDescriptor() function
304 …void CudnnSetTensorNdDescriptor(const std::vector<size_t> &shape, cudnnTensorDescriptor_t descript… in CudnnSetTensorNdDescriptor() function