Searched refs:quantized_img (Results 1 – 1 of 1) sorted by relevance
198 const std::vector<quint8>& quantized_img) { in BuildQuantizedImageTensor() argument201 CHECK_EQ(WIDTH * HEIGHT * DEPTH, quantized_img.size()); in BuildQuantizedImageTensor()202 CHECK_EQ(img_tensor.TotalBytes(), quantized_img.size() * sizeof(quint8)); in BuildQuantizedImageTensor()204 std::memcpy(img_tensor.flat<quint8>().data(), quantized_img.data(), in BuildQuantizedImageTensor()465 std::vector<quint8> quantized_img; in TEST() local466 QuantizeImage(img_floats, &quantized_img); in TEST()467 const Tensor img_tensor = BuildQuantizedImageTensor(quantized_img); in TEST()