Home
last modified time | relevance | path

Searched refs:InputT (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.h113 template <typename FuncT, typename InputT, typename OutputT>
116 const DeviceMemory<InputT> &input,
121 template <typename FuncT, typename InputT, typename OutputT>
123 const DeviceMemory<InputT> &input,
Drocm_fft.cc503 template <typename FuncT, typename InputT, typename OutputT>
505 const DeviceMemory<InputT> &input, in DoFftInternal()
518 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftInternal()
529 template <typename FuncT, typename InputT, typename OutputT>
532 const DeviceMemory<InputT> &input, in DoFftWithDirectionInternal()
545 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.h118 template <typename FuncT, typename InputT, typename OutputT>
121 const DeviceMemory<InputT> &input,
126 template <typename FuncT, typename InputT, typename OutputT>
128 const DeviceMemory<InputT> &input,
Dcuda_fft.cc454 template <typename FuncT, typename InputT, typename OutputT>
456 const DeviceMemory<InputT> &input, in DoFftInternal()
470 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftInternal()
481 template <typename FuncT, typename InputT, typename OutputT>
484 const DeviceMemory<InputT> &input, in DoFftWithDirectionInternal()
498 GpuComplex(const_cast<InputT *>(GpuMemory(input))), in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_functor.h87 using InputT = typename std::conditional<B2S, T, const T>::type; member
105 typename TTypes<InputT, NUM_BLOCK_DIMS + 2>::Tensor space_tensor,
/external/tensorflow/tensorflow/lite/kernels/
Dgather.cc96 template <typename InputT, typename PositionsT>
102 GetTensorData<InputT>(input), GetTensorShape(positions), in Gather()
104 GetTensorShape(output), GetTensorData<InputT>(output)); in Gather()
/external/deqp/modules/gles2/functional/
Des2fShaderMatrixTests.cpp335 template <int InputT, int DataT>
/external/deqp/modules/gles3/functional/
Des3fShaderMatrixTests.cpp493 template <int InputT, int DataT>
/external/deqp/external/vulkancts/modules/vulkan/shaderrender/
DvktShaderRenderMatrixTests.cpp488 template <int InputT, int DataT>