Searched refs:expand_input (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 821 int64 stride, bool expand_input, in GatherToBuffer() argument
825 const int64 ub = expand_input ? length / 2 + 1 : length; in GatherToBuffer()
831 if (expand_input) { in GatherToBuffer()
861 bool contract_output, bool expand_input, in NaiveDft1D() argument
864 GatherToBuffer(data, length, start, stride, expand_input, buffer); in NaiveDft1D()
887 bool contract_output, bool expand_input, absl::Span<complex128> data, in Fft1D() argument
891 GatherToBuffer(data, length, start, stride, expand_input, buffer); in Fft1D()
941 bool contract_output, bool expand_input, absl::Span<complex128> data, in Dft1D() argument
944 Fft1D(length, start, stride, inverse, contract_output, expand_input, data, in Dft1D()
947 NaiveDft1D(length, start, stride, inverse, contract_output, expand_input, in Dft1D()
[all …]
|