Kernels:

phanerozoic
/

dpx-decode

Kernel card Files Files and versions

dpx-decode / torch-ext /torch_binding.h

phanerozoic's picture

dpx-decode v1: kernel-builder source tree

1a42548 verified 27 days ago

538 Bytes

	#pragma once

	#include <torch/torch.h>

	void viterbi(torch::Tensor &path, torch::Tensor &score, torch::Tensor &bp,
	torch::Tensor const &emissions, torch::Tensor const &transitions,
	torch::Tensor const &priors);

	void dtw(torch::Tensor &path, torch::Tensor &path_len, torch::Tensor &D,
	torch::Tensor const &cost);

	void ctc_forced_align(torch::Tensor &frames, torch::Tensor &score,
	torch::Tensor const &log_probs, torch::Tensor const &targets,
	int64_t blank);