Searched refs:row_batch (Results 1 – 1 of 1) sorted by relevance
128 def extract_features(row_batch, col_batch, num_rows, num_cols): argument129 row_ids = row_batch[0]132 row_batch[1], row_ids, shape=[num_rows, num_cols])148 row_batch = input_lib.batch(159 features = extract_features(row_batch, col_batch, num_rows, num_cols)