Searched refs:sweep_axis (Results 1 – 1 of 1) sorted by relevance
1017 std::function<void(int64, int64, int64)> sweep = [&](int64 sweep_axis, in Sweep()1022 const int64 length = fft_lengths[sweep_axis]; in Sweep()1023 const int64 stride = fft_strides[sweep_axis]; in Sweep()1024 const bool expand_input = input_is_truncated && sweep_axis == 0; in Sweep()1025 const bool contract_oputput = output_is_truncated && sweep_axis == 0; in Sweep()1028 } else if (axis == sweep_axis) { in Sweep()1030 sweep(sweep_axis, axis - 1, start); in Sweep()1036 sweep(sweep_axis, axis - 1, start + i * fft_strides[axis]); in Sweep()1042 for (int64 sweep_axis = fft_rank - 1; sweep_axis >= 0; sweep_axis--) { in Sweep() local1043 sweep(sweep_axis, fft_rank - 1, 0); in Sweep()[all …]