Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantize_and_dequantize_op_gpu.cu.cc57 struct QuantizeAndDequantizeOneScaleGradientFunctor<GPUDevice, T> { struct
58 void operator()(const GPUDevice& d, typename TTypes<T>::ConstFlat gradient, in operator ()()
99 template struct functor::QuantizeAndDequantizeOneScaleGradientFunctor<GPUDevice, member in tensorflow::functor
101 template struct functor::QuantizeAndDequantizeOneScaleGradientFunctor<GPUDevice, member in tensorflow::functor
H A Dquantize_and_dequantize_op.cc405 struct QuantizeAndDequantizeOneScaleGradientFunctor<CpuDevice, T> { struct
406 void operator()(const CpuDevice& d, typename TTypes<T>::ConstFlat gradient, in operator ()()
435 template struct functor::QuantizeAndDequantizeOneScaleGradientFunctor<CpuDevice, member in tensorflow::functor::functor
H A Dquantize_and_dequantize_op.h64 struct QuantizeAndDequantizeOneScaleGradientFunctor { struct