Welcome to mirror list, hosted at ThFree Co, Russian Federation.

helpers.h « translator « src - github.com/marian-nmt/marian.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 68159fa31ab34f25a7836529b59fd1d8ff9d8849 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#pragma once

#include "graph/expression_graph.h"

namespace marian {

namespace cpu {

void suppressWords(Expr logProbs, Expr wordIndices);
}

namespace gpu {

void suppressWords(Expr logProbs, Expr wordIndices);
}

void suppressWords(Expr logProbs, Expr wordIndices);
}  // namespace marian