Searched refs:src_lengths (Results 1 – 1 of 1) sorted by relevance
1088 const absl::Span<const int64> src_lengths, in GenerateIndices() argument1093 CHECK_EQ(src_lengths.size() + 1, src_strides.size()); in GenerateIndices()1094 CHECK_GE(src_lengths.size(), fft_rank); in GenerateIndices()1104 within_src_bounds &= i < src_lengths[axis]; in GenerateIndices()