Searched refs:perm_elems (Results 1 – 1 of 1) sorted by relevance
125 DimensionHandle perm_elems = c->NumElements(perm_shape); in TransposeShapeFn() local129 if (!c->RankKnown(input) && !c->ValueKnown(perm_elems) && perm == nullptr) { in TransposeShapeFn()138 } else if (c->ValueKnown(perm_elems)) { in TransposeShapeFn()139 rank = c->Value(perm_elems); in TransposeShapeFn()156 TF_RETURN_IF_ERROR(c->WithValue(perm_elems, rank, &perm_elems)); in TransposeShapeFn()