19 #include "ngraph/op/op.hpp"
30 static constexpr NodeTypeInfo type_info{
"CTCLoss", 0};
31 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
50 const Output<Node>& logit_length,
51 const Output<Node>& labels,
52 const Output<Node>& label_length,
53 const bool preprocess_collapse_repeated =
false,
54 const bool ctc_merge_repeated =
true,
55 const bool unique =
false);
57 CTCLoss(
const Output<Node>& logits,
58 const Output<Node>& logit_length,
59 const Output<Node>& labels,
60 const Output<Node>& label_length,
61 const Output<Node>& blank_index,
62 const bool preprocess_collapse_repeated =
false,
63 const bool ctc_merge_repeated =
true,
64 const bool unique =
false);
68 virtual std::shared_ptr<Node>
69 clone_with_new_inputs(
const OutputVector& new_args)
const override;
71 bool get_preprocess_collapse_repeated()
const
73 return preprocess_collapse_repeated_;
75 bool get_ctc_merge_repeated()
const {
return ctc_merge_repeated_; }
76 bool get_unique()
const {
return unique_; }
78 bool preprocess_collapse_repeated_;
79 bool ctc_merge_repeated_;