Searched defs:output_distance (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_fft.cc | 99 uint64_t *output_embed, uint64 output_stride, uint64 output_distance, in Initialize()
419 uint64_t output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlan()
447 uint64_t output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlanWithScratchAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_fft.cc | 160 uint64_t *output_embed, uint64 output_stride, uint64 output_distance, in Initialize()
466 uint64_t output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlan()
484 uint64_t output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlanWithScratchAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fft_ops.cc | 454 uint64 output_distance = 1; in DoFFT() local
545 const uint64 output_distance, const Tensor& in, in DoFFTInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | fft_thunk.cc | 127 uint64_t output_distance = 1; in RunFft() local
|