Searched refs:params_subshape (Results 1 – 1 of 1) sorted by relevance
1166 ShapeHandle params_subshape; in __anondb9326b21702() local1167 TF_RETURN_IF_ERROR(c->Subshape(c->input(0), 1, ¶ms_subshape)); in __anondb9326b21702()1170 TF_RETURN_IF_ERROR(c->Concatenate(indices_shape, params_subshape, &out)); in __anondb9326b21702()