Searched refs:diag_batch_shape (Results 1 – 1 of 1) sorted by relevance
879 ShapeHandle diag_batch_shape; in __anondb9326b21102() local880 TF_RETURN_IF_ERROR(c->Subshape(diag, 0, -1, &diag_batch_shape)); in __anondb9326b21102()882 c->Concatenate(diag_batch_shape, c->UnknownShapeOfRank(2), &diag)); in __anondb9326b21102()