Searched refs:inferred_dim (Results 1 – 2 of 2) sorted by relevance
1697 DimensionHandle inferred_dim; in __anonb22bfa861402() local1700 &inferred_dim)); in __anonb22bfa861402()1702 c->ReplaceDim(output, rank - 1, inferred_dim, &output)); in __anonb22bfa861402()
244 DimensionHandle inferred_dim; in SetOutputShapeForReshape() local247 &inferred_dim)); in SetOutputShapeForReshape()249 c->ReplaceDim(out, out_unknown_idx, inferred_dim, &out)); in SetOutputShapeForReshape()254 DimensionHandle inferred_dim; in SetOutputShapeForReshape() local257 &inferred_dim)); in SetOutputShapeForReshape()260 c->Merge(unknown_in_dim, inferred_dim, &unknown_in_dim)); in SetOutputShapeForReshape()