Searched refs:dense_rows (Results 1 – 1 of 1) sorted by relevance
76 const int64 dense_rows = dense_shape(0); in Compute() local80 TensorShape({dense_rows}), in Compute()91 if (dense_rows == 0) { in Compute()113 context->allocate_temp(DT_INT64, TensorShape({dense_rows}), in Compute()119 OP_REQUIRES(context, row >= 0 && row < dense_rows, in Compute()121 row, " >= ", dense_rows)); in Compute()124 for (int row = 0; row < dense_rows; ++row) { in Compute()140 const int64 N_full = scratch(dense_rows - 1); in Compute()157 context->allocate_temp(DT_INT64, TensorShape({dense_rows}), in Compute()175 for (int64 row = 0; row < dense_rows; ++row) { in Compute()