Searched refs:dst_lengths (Results 1 – 1 of 1) sorted by relevance
1086 void GenerateIndices(const absl::Span<const int64> dst_lengths, in GenerateIndices() argument1091 CHECK_EQ(dst_lengths.size() + 1, dst_strides.size()); in GenerateIndices()1092 CHECK_GE(dst_lengths.size(), fft_rank); in GenerateIndices()1100 for (int64 i = 0; i < dst_lengths[axis]; i++) { in GenerateIndices()