Searched refs:inferred_dim (Results 1 – 3 of 3) sorted by relevance
1943 DimensionHandle inferred_dim; in __anon57f5a5dd1702() local1946 &inferred_dim)); in __anon57f5a5dd1702()1948 c->ReplaceDim(output, rank - 1, inferred_dim, &output)); in __anon57f5a5dd1702()
250 DimensionHandle inferred_dim; in SetOutputShapeForReshape() local253 &inferred_dim)); in SetOutputShapeForReshape()255 c->ReplaceDim(out, out_unknown_idx, inferred_dim, &out)); in SetOutputShapeForReshape()260 DimensionHandle inferred_dim; in SetOutputShapeForReshape() local263 &inferred_dim)); in SetOutputShapeForReshape()266 c->Merge(unknown_in_dim, inferred_dim, &unknown_in_dim)); in SetOutputShapeForReshape()
1999 int64_t &inferred_dim = (*inferred_item_shape)[i]; in Verify() local2001 if (ShapedType::isDynamic(inferred_dim)) inferred_dim = dim; in Verify()