Searched defs:input_distance (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 78 uint64 *input_embed, uint64 input_stride, uint64 input_distance, in Initialize()
391 uint64 input_stride, uint64 input_distance, uint64 *output_embed, in CreateBatchedPlan()
418 uint64 input_stride, uint64 input_distance, uint64 *output_embed, in CreateBatchedPlanWithScratchAllocator()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_fft.cc | 147 uint64 *input_embed, uint64 input_stride, uint64 input_distance, in Initialize()
464 uint64 input_stride, uint64 input_distance, uint64 *output_embed, in CreateBatchedPlan()
482 uint64 input_stride, uint64 input_distance, uint64 *output_embed, in CreateBatchedPlanWithScratchAllocator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 131 uint64 input_distance = 1; in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fft_ops.cc | 391 uint64 input_distance = 1; in DoFFT() local
|