Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1001 const absl::Span<const int64> fft_strides, in Sweep() argument
1023 const int64 stride = fft_strides[sweep_axis]; in Sweep()
1036 sweep(sweep_axis, axis - 1, start + i * fft_strides[axis]); in Sweep()
1133 const absl::Span<const int64> fft_strides, in CopyDataFromInput() argument
1161 data[dst_index + i * fft_strides[axis]] = value; in CopyDataFromInput()
1167 GenerateIndices(fft_lengths, fft_strides, input_lengths, input_strides, in CopyDataFromInput()
1186 const absl::Span<const int64> fft_strides, in CopyDataToOutput() argument
1207 data[src_index + i * fft_strides[axis]]); in CopyDataToOutput()
1215 GenerateIndices(output_lengths, output_strides, fft_lengths, fft_strides, in CopyDataToOutput()
1223 const absl::Span<const int64> fft_strides, in CopyDataFromInput() argument
[all …]