Searched refs:fft_lengths (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | rfft2d_test.cc | 36 Rfft2dOpModel(const TensorData& input, const TensorData& fft_lengths) { in Rfft2dOpModel() argument
38 fft_lengths_ = AddInput(fft_lengths); in Rfft2dOpModel()
48 int fft_lengths() { return fft_lengths_; } in fft_lengths() function in tflite::ops::builtin::__anon4d2d8d100111::Rfft2dOpModel
70 model.PopulateTensor<int32_t>(model.fft_lengths(), {4, 4}); in TEST()
88 model.PopulateTensor<int32_t>(model.fft_lengths(), {4, 4}); in TEST()
105 model.PopulateTensor<int32_t>(model.fft_lengths(), {4, 8}); in TEST()
131 model.PopulateTensor<int32_t>(model.fft_lengths(), {2, 4}); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1000 const absl::Span<const int64> fft_lengths, in Sweep() argument
1022 const int64 length = fft_lengths[sweep_axis]; in Sweep()
1032 const int64 length = fft_lengths[axis]; in Sweep()
1132 const absl::Span<const int64> fft_lengths, in CopyDataFromInput() argument
1151 const int64 length = fft_lengths[axis]; in CopyDataFromInput()
1167 GenerateIndices(fft_lengths, fft_strides, input_lengths, input_strides, in CopyDataFromInput()
1185 const absl::Span<const int64> fft_lengths, in CopyDataToOutput() argument
1200 const int64 length = fft_lengths[axis]; in CopyDataToOutput()
1215 GenerateIndices(output_lengths, output_strides, fft_lengths, fft_strides, in CopyDataToOutput()
1222 const absl::Span<const int64> fft_lengths, in CopyDataFromInput() argument
[all …]
|