Searched refs:lhs_kept_dim (Results 1 – 1 of 1) sorted by relevance
1154 int64 lhs_kept_dim = kept_dim(lhs_rank, lhs_collapsing_dim, in HandleDotStrengthReduction() local1157 if (lhs_kept_dim == -1 && lhs_rank > 1) { in HandleDotStrengthReduction()1162 std::swap(lhs_collapsing_dim, lhs_kept_dim); in HandleDotStrengthReduction()1240 (lhs_rank == 2 && lhs->shape().dimensions(lhs_kept_dim) == 1)) { in HandleDotStrengthReduction()1303 lhs_kept_dim > lhs_collapsing_dim ? (lhs_kept_dim - 1) : lhs_kept_dim; in HandleDotStrengthReduction()1324 if (lhs->shape().dimensions(lhs_kept_dim) == 1 && in HandleDotStrengthReduction()1326 lhs_kept_dim == rhs_kept_dim) { in HandleDotStrengthReduction()1327 auto new_lhs = StripDim(lhs, lhs_kept_dim); in HandleDotStrengthReduction()1340 if (lhs->shape().dimensions(lhs_kept_dim) == 1) { in HandleDotStrengthReduction()1341 auto new_lhs = broadcast(StripDim(lhs, lhs_kept_dim), rhs->shape(), in HandleDotStrengthReduction()[all …]