Searched refs:seq_dim (Results 1 – 4 of 4) sorted by relevance
461 int seq_dim; member
1418 void ReverseSequence(const TS* seq_lengths, const int seq_dim, in ReverseSequence() argument1425 int outer_dim = std::min(batch_dim, seq_dim); in ReverseSequence()1426 int medium_dim = std::max(batch_dim, seq_dim); in ReverseSequence()1445 if (batch_dim > seq_dim) { in ReverseSequence()1469 } else if (batch_dim < seq_dim) { in ReverseSequence()
727 params->seq_dim = reverse_seq_params->seq_dim(); in ParseOpDataTfLite()
9087 int32_t seq_dim;9090 : seq_dim(0),9101 int32_t seq_dim() const {9121 void add_seq_dim(int32_t seq_dim) {9122 fbb_.AddElement<int32_t>(ReverseSequenceOptions::VT_SEQ_DIM, seq_dim, 0);9141 int32_t seq_dim = 0,9145 builder_.add_seq_dim(seq_dim);14487 { auto _e = seq_dim(); _o->seq_dim = _e; }14499 auto _seq_dim = _o->seq_dim;
Completed in 31 milliseconds