Searched defs:contracting_dimensions (Results 1 – 2 of 2) sorted by relevance
693 DenseIntElementsAttr contracting_dimensions) { in DotDimensionsInfo()716 const DimensionVector &contracting_dimensions() const { in contracting_dimensions() function in mlir::TF::__anon964666f10111::DotDimensionsInfo
434 absl::Span<const int64> contracting_dimensions) { in NormalizeDotOperandToBatchMajorAndContractingMinor()