Searched defs:rhs_rank (Results 1 – 11 of 11) sorted by relevance
161 const int64_t rhs_rank = rhs.size(); in CompareIndices() local
103 const int64_t rhs_rank = rhs_shape.rank(); in CanonicalizeDot() local
1084 int64_t rhs_rank = rhs->shape().rank(); in ComputeArrayForDotWithIndexedRhs() local
70 const auto rhs_rank = rhs_shape.rank(); in CheckSystemAndReturnNumEquations() local126 const int64_t rhs_rank = rhs_shape.rank(); in ValidateTridiagonalMatMulDiagonal() local
146 const int rhs_rank = NumDimensions(rhs); in InitializeTemporaries() local185 int rhs_rank = NumDimensions(rhs); in InitializeTemporaries() local361 const int rhs_rank = NumDimensions(rhs_data); in Prepare() local
286 const int32_t rhs_rank = c->Rank(rhs_shape_handle); in XlaDotShapeFunction() local
515 int rhs_rank = in reshapeForBatchMatmul() local
1678 const int rhs_rank = rhs_type.getRank(); in ConvertDot() local
3313 int64_t rhs_rank = rhs_ty.getRank(); in inferReturnTypeComponents() local
343 const DotConvDimsMapping& dims_mapping, int64_t lhs_rank, int64_t rhs_rank, in ComputeDimensionIndexMapping()
700 const int64_t rhs_rank = rhs_shape->rank(); in BinaryOp() local