Searched refs:GetDimensionNumber (Results 1 – 4 of 4) sorted by relevance
491 ShapeUtil::GetDimensionNumber(lhs_shape, -1); in HandleDot()494 ? ShapeUtil::GetDimensionNumber(rhs_shape, -2) in HandleDot()
296 static int64 GetDimensionNumber(const Shape& shape, int64 dimension_number);
648 return shape.dimensions(GetDimensionNumber(shape, dimension_number)); in GetDimension()651 /* static */ int64 ShapeUtil::GetDimensionNumber(const Shape& shape, in GetDimensionNumber() function in xla::ShapeUtil
744 ShapeUtil::GetDimensionNumber(lhs_shape, transpose_lhs_ ? -2 : -1); in Emit()749 ShapeUtil::GetDimensionNumber(rhs_shape, transpose_rhs_ ? -1 : -2); in Emit()