Searched refs:superdiag_batch_shape (Results 1 – 1 of 1) sorted by relevance
297 ShapeHandle superdiag_batch_shape; in TridiagonalMatMulShapeFn() local301 TF_RETURN_IF_ERROR(c->Subshape(superdiag, 0, -2, &superdiag_batch_shape)); in TridiagonalMatMulShapeFn()