Lines Matching refs:SimpleTensor

37 SimpleTensor<Tout> quantization_layer(const SimpleTensor<Tin> &src, DataType output_data_type, cons…  in quantization_layer()
40     SimpleTensor<Tout> dst{ src.shape(), output_data_type, 1, quantization_info };  in quantization_layer()
90 SimpleTensor<uint8_t> quantization_layer(const SimpleTensor<uint8_t> &src, DataType output_data_typ…  in quantization_layer()
92     SimpleTensor<float> src_tmp = convert_from_asymmetric<uint8_t>(src);  in quantization_layer()
97 SimpleTensor<int8_t> quantization_layer(const SimpleTensor<uint8_t> &src, DataType output_data_type…  in quantization_layer()
99     SimpleTensor<float> src_tmp = convert_from_asymmetric<uint8_t>(src);  in quantization_layer()
104 SimpleTensor<uint8_t> quantization_layer(const SimpleTensor<int8_t> &src, DataType output_data_type…  in quantization_layer()
106     SimpleTensor<float> src_tmp = convert_from_asymmetric<int8_t>(src);  in quantization_layer()
111 SimpleTensor<int8_t> quantization_layer(const SimpleTensor<int8_t> &src, DataType output_data_type,…  in quantization_layer()
113     SimpleTensor<float> src_tmp = convert_from_asymmetric<int8_t>(src);  in quantization_layer()
118 SimpleTensor<uint16_t> quantization_layer(const SimpleTensor<uint8_t> &src, DataType output_data_ty…  in quantization_layer()
120     SimpleTensor<float> src_tmp = convert_from_asymmetric<uint8_t>(src);  in quantization_layer()
124 template SimpleTensor<int8_t> quantization_layer(const SimpleTensor<half> &src, DataType output_dat…
125 template SimpleTensor<int8_t> quantization_layer(const SimpleTensor<float> &src, DataType output_da…
126 template SimpleTensor<uint8_t> quantization_layer(const SimpleTensor<half> &src, DataType output_da…
127 template SimpleTensor<uint8_t> quantization_layer(const SimpleTensor<float> &src, DataType output_d…
128 template SimpleTensor<uint16_t> quantization_layer(const SimpleTensor<half> &src, DataType output_d…
129 template SimpleTensor<uint16_t> quantization_layer(const SimpleTensor<float> &src, DataType output_…