Searched refs:r_dim (Results 1 – 1 of 1) sorted by relevance
2300 DimensionHandle r_dim = c->Dim(indices, -1); in GatherNdShape() local2302 if (!c->RankKnown(params) || !c->ValueKnown(r_dim)) { in GatherNdShape()2307 if (c->Value(r_dim) > c->Rank(params)) { in GatherNdShape()2317 TF_RETURN_IF_ERROR(c->Subshape(params, c->Value(r_dim), ¶ms_slice)); in GatherNdShape()