Home
last modified time | relevance | path

Searched refs:OutputT (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.h113 template <typename FuncT, typename InputT, typename OutputT>
117 DeviceMemory<OutputT> *output);
121 template <typename FuncT, typename InputT, typename OutputT>
124 DeviceMemory<OutputT> *output);
Drocm_fft.cc503 template <typename FuncT, typename InputT, typename OutputT>
506 DeviceMemory<OutputT> *output) { in DoFftInternal()
529 template <typename FuncT, typename InputT, typename OutputT>
533 DeviceMemory<OutputT> *output) { in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.h118 template <typename FuncT, typename InputT, typename OutputT>
122 DeviceMemory<OutputT> *output);
126 template <typename FuncT, typename InputT, typename OutputT>
129 DeviceMemory<OutputT> *output);
Dcuda_fft.cc454 template <typename FuncT, typename InputT, typename OutputT>
457 DeviceMemory<OutputT> *output) { in DoFftInternal()
481 template <typename FuncT, typename InputT, typename OutputT>
485 DeviceMemory<OutputT> *output) { in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_functor.h88 using OutputT = typename std::conditional<B2S, const T, T>::type; member
108 typename TTypes<OutputT, NUM_BLOCK_DIMS + 2>::Tensor batch_tensor);