Searched refs:contracting_dimensions (Results 1 – 2 of 2) sorted by relevance
396 DenseIntElementsAttr contracting_dimensions) { in DotDimensionsInfo() argument403 for (const int dim : contracting_dimensions.getValues<int64_t>()) { in DotDimensionsInfo()419 const DimensionVector &contracting_dimensions() const { in contracting_dimensions() function in mlir::TF::__anon0527096c0111::DotDimensionsInfo470 lhs_dot_dimensions_info.contracting_dimensions().AxesArray()); in ConvertDot()474 lhs_dot_dimensions_info.contracting_dimensions().SizesArray()); in ConvertDot()487 rhs_dot_dimensions_info.contracting_dimensions().AxesArray(), in ConvertDot()491 rhs_dot_dimensions_info.contracting_dimensions().SizesArray(), in ConvertDot()
398 absl::Span<const int64> contracting_dimensions) { in NormalizeDotOperandToBatchMajorAndContractingMinor() argument403 absl::c_linear_search(contracting_dimensions, i))) { in NormalizeDotOperandToBatchMajorAndContractingMinor()408 contracting_dimensions.begin(), in NormalizeDotOperandToBatchMajorAndContractingMinor()409 contracting_dimensions.end()); in NormalizeDotOperandToBatchMajorAndContractingMinor()