Searched defs:CTCLossDescriptor (Results 1 – 1 of 1) sorted by relevance
343 struct TORCH_CUDA_CPP_API CTCLossDescriptor struct349 AT_CUDNN_CHECK(cudnnSetCTCLossDescriptor(mut_desc(), datatype)); in set() argument356 cudnnSetCTCLossDescriptorEx(mut_desc(), datatype, normMode, gradMode)); in setEx() argument369 cudnnSetCTCLossDescriptor_v9(mut_desc(), datatype, normMode, gradModev9, maxLabelLength)); in set_v8_v9() argument