Searched refs:kSeqDim (Results 1 – 2 of 2) sorted by relevance
30 void ReverseSequence::set_seq_dim(const int64_t seq_dim) { (void)this->AddAttr(kSeqDim, MakeValue(s… in set_seq_dim()33 int64_t ReverseSequence::get_seq_dim() const { return GetValue<int64_t>(GetAttr(kSeqDim)); } in get_seq_dim()51 auto seq_dim = GetValue<int64_t>(primitive->GetAttr(kSeqDim)); in ReverseSequenceInfer()
177 constexpr auto kSeqDim = "seq_dim"; variable