Searched refs:rhs_contracting_dimension (Results 1 – 4 of 4) sorted by relevance
200 const int64 rhs_contracting_dimension = in __anonbbc1c2d10202() local207 c->DimKnownRank(rhs_shape_handle, rhs_contracting_dimension)); in __anonbbc1c2d10202()
1351 const int64 rhs_contracting_dimension = dnums.rhs_contracting_dimensions(0);1354 rhs->shape().dimensions(rhs_contracting_dimension))1358 << rhs->shape().dimensions(rhs_contracting_dimension);1362 rhs_contracting_dimension == 0 &&
2005 int rhs_contracting_dimension = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfGather() local2010 rhs, lhs, /*a_contracting_dimension=*/rhs_contracting_dimension)) { in OptimizeDotOfGather()2045 right_operand->shape().dimensions(1 - rhs_contracting_dimension); in OptimizeDotOfGather()2056 : 1 - rhs_contracting_dimension; in OptimizeDotOfGather()
680 const int64 rhs_contracting_dimension = in InferDotOpShape() local683 rhs.dimensions(rhs_contracting_dimension)) { in InferDotOpShape()