Home
last modified time | relevance | path

Searched defs:output_distance (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.cc160 uint64 *output_embed, uint64 output_stride, uint64 output_distance, in Initialize()
464 uint64 output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlan()
482 uint64 output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlanWithScratchAllocator()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.cc80 uint64 *output_embed, uint64 output_stride, uint64 output_distance, in Initialize()
395 uint64 output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlan()
422 uint64 output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlanWithScratchAllocator()
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc443 uint64 output_distance = 1; in DoFFT() local
530 const uint64 output_distance, const Tensor& in, in DoFFTInternal()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.cc158 uint64 output_distance = 1; in ExecuteOnStream() local