Home
last modified time | relevance | path

Searched refs:tensor_failed_target_lengths_check (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DLossCTC.cpp86 static bool tensor_failed_target_lengths_check = false; variable
147 tensor_failed_target_lengths_check = true; in _use_cudnn_ctc_loss_tensor()
152 use_cudnn = use_cudnn && !tensor_failed_target_lengths_check; in _use_cudnn_ctc_loss_tensor()
153 if (tensor_failed_target_lengths_check) { in _use_cudnn_ctc_loss_tensor()