Searched defs:contracting_dimensions (Results 1 – 2 of 2) sorted by relevance
396 DenseIntElementsAttr contracting_dimensions) { in DotDimensionsInfo()419 const DimensionVector &contracting_dimensions() const { in contracting_dimensions() function in mlir::TF::__anon0527096c0111::DotDimensionsInfo
398 absl::Span<const int64> contracting_dimensions) { in NormalizeDotOperandToBatchMajorAndContractingMinor()