Searched refs:num_row_splits (Results 1 – 1 of 1) sorted by relevance
309 DimensionHandle num_row_splits; in __anonf38d6eaa0902() local311 TF_RETURN_IF_ERROR(c->Add(input_size, 1, &num_row_splits)); in __anonf38d6eaa0902()312 c->set_output(0, c->Vector(num_row_splits)); in __anonf38d6eaa0902()331 DimensionHandle num_row_splits; in __anonf38d6eaa0a02() local333 TF_RETURN_IF_ERROR(c->Add(input_size, 1, &num_row_splits)); in __anonf38d6eaa0a02()334 c->set_output(0, c->Vector(num_row_splits)); in __anonf38d6eaa0a02()