Searched refs:lhs_contracting_dimension (Results 1 – 4 of 4) sorted by relevance
198 const int64 lhs_contracting_dimension = in __anonbbc1c2d10202() local204 c->DimKnownRank(lhs_shape_handle, lhs_contracting_dimension)); in __anonbbc1c2d10202()
1350 const int64 lhs_contracting_dimension = dnums.lhs_contracting_dimensions(0);1353 CHECK_EQ(lhs->shape().dimensions(lhs_contracting_dimension),1356 << lhs->shape().dimensions(lhs_contracting_dimension)1361 if (lhs_rank == 2 && rhs_rank == 2 && lhs_contracting_dimension == 1 &&1372 lhs->shape().dimensions(lhs_contracting_dimension);
2004 int lhs_contracting_dimension = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfGather() local2008 lhs, rhs, /*a_contracting_dimension=*/lhs_contracting_dimension) && in OptimizeDotOfGather()2043 const int m = left_operand->shape().dimensions(1 - lhs_contracting_dimension); in OptimizeDotOfGather()2055 ? 1 - lhs_contracting_dimension in OptimizeDotOfGather()
678 const int64 lhs_contracting_dimension = in InferDotOpShape() local682 if (lhs.dimensions(lhs_contracting_dimension) != in InferDotOpShape()