Searched refs:lhs_batch_dimension (Results 1 – 1 of 1) sorted by relevance
263 const int64_t lhs_batch_dimension = in XlaDotShapeFunction() local269 c->Merge(c->DimKnownRank(lhs_shape_handle, lhs_batch_dimension), in XlaDotShapeFunction()272 lhs_batch_dimension, " and rhs dimension ", rhs_batch_dimension); in XlaDotShapeFunction()