Searched refs:output_row_splits (Results 1 – 1 of 1) sorted by relevance
419 Tensor* output_row_splits; in Compute() local422 &output_row_splits)); in Compute()423 auto out_row_splits = output_row_splits->vec<int64>(); in Compute()