Searched refs:output_after (Results 1 – 1 of 1) sorted by relevance
1256 ShapeHandle output_after; in ConcatShapeHelper() local1263 output_after = c->Scalar(); // no dimensions. in ConcatShapeHelper()1265 TF_RETURN_IF_ERROR(c->Subshape(input, concat_dim + 1, &output_after)); in ConcatShapeHelper()1283 TF_RETURN_IF_ERROR(c->Merge(after, output_after, &output_after)); in ConcatShapeHelper()1289 TF_RETURN_IF_ERROR(c->Concatenate(s, output_after, &s)); in ConcatShapeHelper()