Searched defs:seq_dim (Results 1 – 12 of 12) sorted by relevance
44 void CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) { in CheckErrors()79 void CheckErrorsGPU(OpKernelContext* context, int batch_dim, int seq_dim) { in CheckErrorsGPU()101 int seq_dim) { in CheckErrors()107 int seq_dim) { in CheckErrors()
33 int32_t seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) in ReverseGenerator()
82 int seq_dim = params->seq_dim; in ReverseSequenceImpl() local
33 int seq_dim, int batch_dim) { in ReverseSequenceOpModel()
271 int seq_dim; in ReverseSequenceGrad() local
1157 const int32_t batch_dim) { in TEST()
1642 int64_t seq_dim; in __anon38bbb0e82402() local
463 int seq_dim; member
475 int seq_dim; member
915 void ReverseSequence(const TS* seq_lengths, const int seq_dim, in ReverseSequence()
4984 seq_dim=None, argument
3934 int seq_dim = shuffled_dim_ids[1]; in TEST_F() local