Searched defs:rhs_rank (Results 1 – 5 of 5) sorted by relevance
221 const int64 rhs_rank = rhs_shape.rank(); in CanonicalizeDot() local
1032 const auto rhs_rank = rhs_shape.rank(); in HandleConvolution() local1224 const int64 rhs_rank = rhs->shape().rank(); in HandleDot() local1285 const auto rhs_rank = rhs->shape().rank(); in HandleDotSlowPath() local
1080 int64 rhs_rank = rhs->shape().rank(); in ComputeArrayForDotWithIndexedRhs() local
1144 const int64 rhs_rank = rhs->shape().rank(); in HandleDotStrengthReduction() local
506 const int64 rhs_rank = rhs_shape.rank(); in BinaryOp() local