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