Searched refs:next_stamp_token_t (Results 1 – 3 of 3) sorted by relevance
567 const Tensor* next_stamp_token_t; in Compute() local569 context->input(kNextStampTokenName, &next_stamp_token_t)); in Compute()570 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute()604 const Tensor* next_stamp_token_t; in Compute() local606 context->input(kNextStampTokenName, &next_stamp_token_t)); in Compute()607 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute()
212 const Tensor* next_stamp_token_t; in Compute() local214 context->input("next_stamp_token", &next_stamp_token_t)); in Compute()215 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute()338 const Tensor* next_stamp_token_t; in Compute() local340 context->input("next_stamp_token", &next_stamp_token_t)); in Compute()341 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute()
606 const Tensor* next_stamp_token_t; in Compute() local608 context->input(kNextStampTokenName, &next_stamp_token_t)); in Compute()609 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute()652 const Tensor* next_stamp_token_t; in Compute() local654 context->input(kNextStampTokenName, &next_stamp_token_t)); in Compute()655 int64 next_stamp_token = next_stamp_token_t->scalar<int64>()(); in Compute()