Searched refs:axis_before_ellipsis (Results 1 – 1 of 1) sorted by relevance
212 const int64_t axis_before_ellipsis = label_idx; in EinsumShape() local225 TF_RETURN_IF_ERROR(c->Subshape(input_shape, axis_before_ellipsis, in EinsumShape()234 int64_t axis = seen_ellipsis ? axis_after_ellipsis : axis_before_ellipsis; in EinsumShape()