Searched defs:QuantizeAndDequantizePerChannelFunctor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantize_and_dequantize_op_gpu.cu.cc | 44 struct QuantizeAndDequantizePerChannelFunctor<GPUDevice, T> { struct
45 void operator()(const GPUDevice& d, typename TTypes<T, 3>::ConstTensor input, in operator ()()
94 template struct functor::QuantizeAndDequantizePerChannelFunctor<GPUDevice, member in tensorflow::functor
96 template struct functor::QuantizeAndDequantizePerChannelFunctor<GPUDevice, member in tensorflow::functor
|
| H A D | quantize_and_dequantize_op.cc | 392 struct QuantizeAndDequantizePerChannelFunctor<CpuDevice, T> { struct
393 void operator()(const CpuDevice& d, typename TTypes<T, 3>::ConstTensor input, in operator ()()
|
| H A D | quantize_and_dequantize_op.h | 55 struct QuantizeAndDequantizePerChannelFunctor { struct
|