Home
last modified time | relevance | path

Searched refs:QUANTIZE_WITH_EIGEN (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dquantization_utils.h157 #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()
Dquantized_concat_op.cc55 auto input_requantized = QUANTIZE_WITH_EIGEN(input_float, f2q, T); in Copy()
Dquantized_instance_norm.cc386 QUANTIZE_WITH_EIGEN(instance_normed, output_f2q, quint8); in Compute()
Dquantization_utils_test.cc671 output.flat<qint32>() = QUANTIZE_WITH_EIGEN(input_array, f2q, qint32); in TestOverflowWithEigen()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_concat_op.cc80 output_array = QUANTIZE_WITH_EIGEN(input_float, f2q, T); in Copy()