Searched defs:contracting_dimensions (Results 1 – 2 of 2) sorted by relevance
1619 ArrayRef<int64_t> contracting_dimensions) { in DotDimensionsInfo()1642 const DimensionVector &contracting_dimensions() const { in contracting_dimensions() function in mlir::TF::__anon09b86ffb0111::DotDimensionsInfo
2093 absl::Span<const int64_t> contracting_dimensions) { in NormalizeDotOperandToBatchMajorAndContractingMinor()