Searched defs:contracting_dims (Results 1 – 2 of 2) sorted by relevance
976 int64 rank, absl::Span<const int64> contracting_dims, in GetOnlyNonContractingNonBatchDim()1001 absl::Span<const int64> contracting_dims, in CanFoldDotIntoIndexedArray()
563 absl::Span<const int64> batch_dims) -> bool { in ValidateDotDimensionNumbers()588 absl::Span<const int64> batch_dims) -> bool { in ValidateDotDimensionNumbers()