Searched defs:output_distance (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_fft.cc | 79 uint64 *output_embed, uint64 output_stride, uint64 output_distance, in Initialize() 392 uint64 output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlan() 419 uint64 output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlanWithScratchAllocator()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_fft.cc | 148 uint64 *output_embed, uint64 output_stride, uint64 output_distance, in Initialize() 465 uint64 output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlan() 483 uint64 output_stride, uint64 output_distance, fft::Type type, in CreateBatchedPlanWithScratchAllocator()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | fft_thunk.cc | 134 uint64 output_distance = 1; in ExecuteOnStream() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | fft_ops.cc | 394 uint64 output_distance = 1; in DoFFT() local
|