Home
last modified time | relevance | path

Searched refs:src_lengths (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1088 const absl::Span<const int64> src_lengths, in GenerateIndices() argument
1093 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()