Searched refs:row_seq_tensor (Results 1 – 1 of 1) sorted by relevance
245 const Tensor& row_seq_tensor = context->input(2); in Compute() local253 OP_REQUIRES(context, row_seq_tensor.NumElements() > out_rows, in Compute()258 row_seq_tensor.NumElements())); in Compute()266 auto row_seq_tensor_flat = row_seq_tensor.flat<int64>(); in Compute()