Searched refs:rhs_contracting_dimension (Results 1 – 4 of 4) sorted by relevance
198 const int64 rhs_contracting_dimension = in __anon00ae58f50202() local205 c->DimKnownRank(rhs_shape_handle, rhs_contracting_dimension)); in __anon00ae58f50202()
1259 const int64 rhs_contracting_dimension = dnums.rhs_contracting_dimensions(0);1262 rhs->shape().dimensions(rhs_contracting_dimension))1266 << rhs->shape().dimensions(rhs_contracting_dimension);1270 rhs_contracting_dimension == 0 &&
1549 int rhs_contracting_dimension = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfGather() local1554 rhs, lhs, /*a_contracting_dimension=*/rhs_contracting_dimension)) { in OptimizeDotOfGather()1589 right_operand->shape().dimensions(1 - rhs_contracting_dimension); in OptimizeDotOfGather()1600 : 1 - rhs_contracting_dimension; in OptimizeDotOfGather()
663 const int64 rhs_contracting_dimension = in InferDotOpShape() local666 rhs.dimensions(rhs_contracting_dimension)) { in InferDotOpShape()