Searched refs:QUANTIZE_WITH_EIGEN (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | quantization_utils.h | 157 #define QUANTIZE_WITH_EIGEN(input_array, f2q, OutputType) \ macro 685 auto input_requantized = QUANTIZE_WITH_EIGEN(input_float, f2q, T2); in RequantizeManyInNewRangeUsingEigen() 746 flat_result.device(device) = QUANTIZE_WITH_EIGEN(flat_input, f2q, T); in FloatTensorToQuantizedInPlaceUsingEigen() 843 QUANTIZE_WITH_EIGEN(smaller_input_float, f2q, T3); in QuantizedAddUsingEigen() 846 auto input_in_total_space = QUANTIZE_WITH_EIGEN(input_float, f2q, T3); in QuantizedAddUsingEigen()
|
D | quantized_concat_op.cc | 55 auto input_requantized = QUANTIZE_WITH_EIGEN(input_float, f2q, T); in Copy()
|
D | quantized_instance_norm.cc | 386 QUANTIZE_WITH_EIGEN(instance_normed, output_f2q, quint8); in Compute()
|
D | quantization_utils_test.cc | 671 output.flat<qint32>() = QUANTIZE_WITH_EIGEN(input_array, f2q, qint32); in TestOverflowWithEigen()
|
/external/tensorflow/tensorflow/core/kernels/mkl/ |
D | mkl_concat_op.cc | 80 output_array = QUANTIZE_WITH_EIGEN(input_float, f2q, T); in Copy()
|