Searched defs:lhs_rank (Results 1 – 9 of 9) sorted by relevance
49 const int64_t lhs_rank = lhs_shape.rank(); in CanonicalizeDot() local
1049 int64_t lhs_rank = lhs->shape().rank(); in ComputeArrayForDotWithIndexedLhs() local
145 const int lhs_rank = NumDimensions(lhs); in InitializeTemporaries() local360 const int lhs_rank = NumDimensions(lhs_data); in Prepare() local
276 const int32_t lhs_rank = c->Rank(lhs_shape_handle); in XlaDotShapeFunction() local
511 int lhs_rank = in reshapeForBatchMatmul() local
1677 const int lhs_rank = lhs_type.getRank(); in ConvertDot() local1767 auto lhs_rank = dot_op.lhs().getType().cast<ShapedType>().getRank(); in ConvertDotOp() local
3312 int64_t lhs_rank = lhs_ty.getRank(); in inferReturnTypeComponents() local
343 const DotConvDimsMapping& dims_mapping, int64_t lhs_rank, int64_t rhs_rank, in ComputeDimensionIndexMapping()
699 const int64_t lhs_rank = lhs_shape->rank(); in BinaryOp() local