Searched refs:perm_elems (Results 1 – 1 of 1) sorted by relevance
118 DimensionHandle perm_elems = c->NumElements(perm_shape); in TransposeShapeFn() local122 if (!c->RankKnown(input) && !c->ValueKnown(perm_elems) && perm == nullptr) { in TransposeShapeFn()131 } else if (c->ValueKnown(perm_elems)) { in TransposeShapeFn()132 rank = c->Value(perm_elems); in TransposeShapeFn()149 TF_RETURN_IF_ERROR(c->WithValue(perm_elems, rank, &perm_elems)); in TransposeShapeFn()