Searched defs:input_embed (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 | 98 uint64_t *input_embed, uint64 input_stride, uint64 input_distance, in Initialize()
417 Stream *stream, int rank, uint64_t *elem_count, uint64 *input_embed, in CreateBatchedPlan()
445 Stream *stream, int rank, uint64_t *elem_count, uint64 *input_embed, in CreateBatchedPlanWithScratchAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_fft.cc | 159 uint64_t *input_embed, uint64 input_stride, uint64 input_distance, in Initialize()
464 Stream *stream, int rank, uint64_t *elem_count, uint64 *input_embed, in CreateBatchedPlan()
482 Stream *stream, int rank, uint64_t *elem_count, uint64 *input_embed, in CreateBatchedPlanWithScratchAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | fft_thunk.cc | 122 uint64_t input_embed[3]; in RunFft() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fft_ops.cc | 449 uint64 input_embed[3]; in DoFFT() local
|