Searched defs:lhs_rank (Results 1 – 9 of 9) sorted by relevance
47 const int64 lhs_rank = lhs_shape.rank(); in CanonicalizeDot() local
1045 int64 lhs_rank = lhs->shape().rank(); in ComputeArrayForDotWithIndexedLhs() local
260 const int32 lhs_rank = c->Rank(lhs_shape_handle); in __anonbbc1c2d10202() local
145 const int lhs_rank = NumDimensions(lhs); in InitializeTemporaries() local356 const int lhs_rank = NumDimensions(lhs_data); in Prepare() local
454 const int lhs_rank = lhs_type.getRank(); in ConvertDot() local544 auto lhs_rank = dot_op.lhs().getType().cast<ShapedType>().getRank(); in ConvertDotOp() local
421 Value lhs_rank = in HandleBroadcastAndOp() local
340 const DotConvDimsMapping& dims_mapping, int64 lhs_rank, int64 rhs_rank, in ComputeDimensionIndexMapping()
3119 int64_t lhs_rank = lhs_ty.getRank(); in inferReturnTypes() local
636 const int64 lhs_rank = lhs_shape->rank(); in BinaryOp() local