Searched refs:fft_strides (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_evaluator.cc | 1001 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 …]
|