Searched refs:num_row_splits (Results 1 – 1 of 1) sorted by relevance
335 DimensionHandle num_row_splits; in __anonf38d6eaa0902() local337 TF_RETURN_IF_ERROR(c->Add(input_size, 1, &num_row_splits)); in __anonf38d6eaa0902()338 c->set_output(0, c->Vector(num_row_splits)); in __anonf38d6eaa0902()358 DimensionHandle num_row_splits; in __anonf38d6eaa0a02() local360 TF_RETURN_IF_ERROR(c->Add(input_size, 1, &num_row_splits)); in __anonf38d6eaa0a02()361 c->set_output(0, c->Vector(num_row_splits)); in __anonf38d6eaa0a02()