Searched refs:contracting_dims (Results 1 – 2 of 2) sorted by relevance
976 int64 rank, absl::Span<const int64> contracting_dims, in GetOnlyNonContractingNonBatchDim() argument980 if (!absl::c_linear_search(contracting_dims, dim) && in GetOnlyNonContractingNonBatchDim()1001 absl::Span<const int64> contracting_dims, in CanFoldDotIntoIndexedArray() argument1005 contracting_dims, batch_dims); in CanFoldDotIntoIndexedArray()
562 absl::Span<const int64> contracting_dims, in ValidateDotDimensionNumbers()565 return absl::c_all_of(contracting_dims, in_range) && in ValidateDotDimensionNumbers()587 auto dims_unique = [](absl::Span<const int64> contracting_dims, in ValidateDotDimensionNumbers()593 return absl::c_all_of(contracting_dims, is_unique) && in ValidateDotDimensionNumbers()