We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d17635f commit e123269Copy full SHA for e123269
src/libtorchaudio/rnnt/cpu/compute.cpp
@@ -21,7 +21,7 @@ std::tuple<Tensor, Tensor> compute(
21
Tensor target_lengths,
22
int64_t blank,
23
double clamp,
24
- bool fused_log_softmax = true) {
+ bool fused_log_softmax) {
25
STD_TORCH_CHECK(logits.is_cpu(), "logits must be on CPU");
26
27
STD_TORCH_CHECK(
src/libtorchaudio/rnnt/gpu/compute.cu
STD_TORCH_CHECK(logits.is_cuda(), "logits must be on CUDA");
0 commit comments