Functions — github.com/HawkAaron/warp-transducer

CodeHub MCPcopy Create free account

hub / github.com/HawkAaron/warp-transducer / functions

Functions89 in github.com/HawkAaron/warp-transducer

⨍ Functions 89 ◇ Types & classes 22

↓ 22 callersFunctionlogp

include/detail/gpu_rnnt_kernel.h:6

↓ 12 callersFunctionthrow_on_error

tests/test.h:11

↓ 10 callersFunctioncompute_rnnt_loss

src/rnnt_entrypoint.cpp:38

↓ 8 callersFunctionget_workspace_size

src/rnnt_entrypoint.cpp:103

↓ 6 callersFunctiongenActs

tests/random.cpp:4

↓ 4 callersFunctionlog_softmax

pytorch_binding/test/transducer_np.py:8

↓ 4 callersFunctionshow

pytorch_binding/test/test.py:74

↓ 3 callersMethodbackward

(ctx, grad_output)

pytorch_binding/test/transducer_np.py:114

↓ 3 callersFunctiongenLabels

tests/random.cpp:22

↓ 3 callersFunctionrnntGetStatusString

src/rnnt_entrypoint.cpp:18

↓ 3 callersFunctionrnnt_loss

Computes the RNNT loss between a sequence of activations and a ground truth labeling. Args: trans_acts: A 3-D Tensor of floats. The d

tensorflow_binding/warprnnt_tensorflow/__init__.py:9

↓ 2 callersMethod_run_rnnt

(self, trans_acts, pred_acts, labels, input_lengths, label_lengths, ex

tensorflow_binding/tests/test_warprnnt_op.py:12

↓ 2 callersMethod_test_basic

(self, use_gpu)

tensorflow_binding/tests/test_warprnnt_op.py:69

↓ 2 callersMethod_test_multiple_batches

(self, use_gpu)

tensorflow_binding/tests/test_warprnnt_op.py:108

↓ 2 callersMethodcost_and_grad

include/detail/cpu_rnnt.h:289

↓ 2 callersFunctionis_gpu_available

Returns whether Tensorflow can access a GPU.

tensorflow_binding/tests/test_warprnnt_op.py:6

↓ 2 callersFunctionnumeric_grad

tests/test_cpu.cpp:256

↓ 2 callersFunctionrel_diff

tests/test.h:22

↓ 2 callersMethodscore_forward

include/detail/cpu_rnnt.h:328

↓ 1 callersFunctionbackward_pass

(log_probs, labels, blank)

pytorch_binding/test/transducer_np.py:31

↓ 1 callersFunctionbig_test

pytorch_binding/test/test.py:82

↓ 1 callersFunctioncompute_gradient

(log_probs, alphas, betas, labels, blank)

pytorch_binding/test/transducer_np.py:51

↓ 1 callersFunctionforward_pass

(log_probs, labels, blank)

pytorch_binding/test/transducer_np.py:12

↓ 1 callersFunctionget_warprnnt_version

src/rnnt_entrypoint.cpp:14

↓ 1 callersFunctiongrad_check

tests/test_cpu.cpp:300

↓ 1 callersFunctioninf_test

tests/test_cpu.cpp:195

↓ 1 callersMethodlog_softmax

include/detail/cpu_rnnt.h:144

↓ 1 callersFunctionoptions_test

tests/test_cpu.cpp:69

↓ 1 callersFunctionrun_test

tests/test_time.cpp:15

↓ 1 callersFunctionrun_tests

tests/test_cpu.cpp:364

↓ 1 callersFunctionsmall_test

pytorch_binding/test/test.py:34

↓ 1 callersFunctionsmall_test

tests/test_cpu.cpp:12

↓ 1 callersFunctiontransduce

Args: log_probs: 3D array with shape [input len, output len + 1, vocab size] labels: 1D array with shape [output ti

pytorch_binding/test/transducer_np.py:66

↓ 1 callersFunctiontransduce_batch

(probs, labels, flen, glen, blank=0)

pytorch_binding/test/transducer_np.py:82

tensorflow_binding/src/warprnnt_op.cc:34

include/detail/cpu_rnnt.h:21

MethodCpuRNNT_index

include/detail/cpu_rnnt.h:122

MethodCpuRNNT_logProbs

include/detail/cpu_rnnt.h:136

MethodCpuRNNT_metadata

include/detail/cpu_rnnt.h:111

include/detail/gpu_rnnt.h:22

MethodWarpRNNTOpBase

tensorflow_binding/src/warprnnt_op.cc:30

MethodWarpRNNTOpCPU

tensorflow_binding/src/warprnnt_op.cc:159

MethodWarpRNNTOpGPU

tensorflow_binding/src/warprnnt_op.cc:181

Function_RNNTLossGrad

(op, grad_loss, trans_grad, pred_grad)

tensorflow_binding/warprnnt_tensorflow/__init__.py:42

Function_RNNTLossShape

tensorflow_binding/warprnnt_tensorflow/__init__.py:50

(self, size_average=False, blank_label=0, batch_first=True)

pytorch_binding/warprnnt_pytorch/__init__.py:66

(self, blank_label=0, batch_first=True)

pytorch_binding/test/transducer_np.py:124

(ctx, grad_output)

pytorch_binding/warprnnt_pytorch/__init__.py:55

Methodbuild_extensions

tensorflow_binding/setup.py:107

Methodcompute_alphas

include/detail/cpu_rnnt.h:214

Functioncompute_alphas_kernel

include/detail/gpu_rnnt_kernel.h:11

Methodcompute_betas_and_grad

include/detail/cpu_rnnt.h:240

Functioncompute_betas_kernel

include/detail/gpu_rnnt_kernel.h:41

Methodcompute_cost_and_score

include/detail/gpu_rnnt.h:88

Functioncompute_grad_kernel

include/detail/gpu_rnnt_kernel.h:70

Methodcost_and_grad

include/detail/gpu_rnnt.h:177

Methodcost_and_grad_kernel

include/detail/cpu_rnnt.h:176

Functioncpu_rnnt

pytorch_binding/src/binding.cpp:17

Methodcreate_options

tensorflow_binding/src/warprnnt_op.cc:167

Methodcreate_options

tensorflow_binding/src/warprnnt_op.cc:189

Functiondiscover_test_suite

tensorflow_binding/setup.py:111

include/detail/rnnt_helper.h:26

(ctx, trans_acts, pred_acts, labels, act_lens, label_lens, size_average, blank_label)

pytorch_binding/warprnnt_pytorch/__init__.py:13

trans_acts: Tensor of (batch x seqLength x outputDim) containing output from transcription network pred_acts: Tensor of (batch x labe

pytorch_binding/warprnnt_pytorch/__init__.py:73

(ctx, trans_acts, pred_acts, labels, act_lens, label_lens, blank)

pytorch_binding/test/transducer_np.py:97

(self, trans_acts, pred_acts, labels, act_lens, label_lens)

pytorch_binding/test/transducer_np.py:130

Functiongenerate_sample

pytorch_binding/test/test.py:7

Functiongpu_rnnt

pytorch_binding/src/binding.cpp:76

Methodlog_softmax

include/detail/gpu_rnnt.h:79

Functionlog_sum_exp

include/detail/rnnt_helper.h:17

tests/test_time.cpp:92

tests/test_cpu.cpp:397

Functionneg_inf

include/detail/rnnt_helper.h:14

Methodoperator()

include/detail/cpu_rnnt.h:126

Methodoperator()

include/detail/rnnt_helper.h:31

Methodoperator()

include/detail/rnnt_helper.h:38

Methodoperator()

include/detail/rnnt_helper.h:45

Methodoperator()

include/detail/rnnt_helper.h:49

Methodoperator()

include/detail/rnnt_helper.h:53

Methodoperator()

include/detail/rnnt_helper.h:59

Methodscore_forward

include/detail/gpu_rnnt.h:202

tensorflow_binding/src/warprnnt_op.cc:163

tensorflow_binding/src/warprnnt_op.cc:185

Functionsoftmax

Numerically stable softmax for a minibatch of 1

tests/test.h:35

Methodtest_basic_cpu

tensorflow_binding/tests/test_warprnnt_op.py:99

Methodtest_basic_gpu

tensorflow_binding/tests/test_warprnnt_op.py:102

Methodtest_forward

tensorflow_binding/tests/test_warprnnt_op.py:46

Methodtest_multiple_batches_cpu

tensorflow_binding/tests/test_warprnnt_op.py:151

Methodtest_multiple_batches_gpu

tensorflow_binding/tests/test_warprnnt_op.py:154