Searched refs:to_sizes (Results 1 – 2 of 2) sorted by relevance
357 absl::Span<const int64_t> from_sizes, absl::Span<const int64_t> to_sizes) { in ConvertDimensionNumbers() argument359 auto common_factors = CommonFactors(from_sizes, to_sizes); in ConvertDimensionNumbers()388 const int64_t to_size = to_sizes[to_dim]; in ConvertDimensionNumbers()
607 absl::Span<const int64_t> from_sizes, absl::Span<const int64_t> to_sizes);