Home
last modified time | relevance | path

Searched defs:QuantizeAndDequantizeOneScaleFunctor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantize_and_dequantize_op_gpu.cu.cc31 struct QuantizeAndDequantizeOneScaleFunctor<GPUDevice, T> { struct
32 void operator()(const GPUDevice& d, typename TTypes<T>::ConstVec input, in operator ()()
90 template struct functor::QuantizeAndDequantizeOneScaleFunctor<GPUDevice, float>; member in tensorflow::functor
91 template struct functor::QuantizeAndDequantizeOneScaleFunctor<GPUDevice, member in tensorflow::functor
H A Dquantize_and_dequantize_op.cc379 struct QuantizeAndDequantizeOneScaleFunctor<CpuDevice, T> { struct
380 void operator()(const CpuDevice& d, typename TTypes<T>::ConstVec input, in operator ()()
H A Dquantize_and_dequantize_op.h46 struct QuantizeAndDequantizeOneScaleFunctor { struct