Searched refs:img_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_graph_execution_test.cc | 188 Tensor img_tensor(DT_FLOAT, {1, WIDTH, HEIGHT, DEPTH}); in BuildImageTensor() local
190 CHECK_EQ(img_tensor.TotalBytes(), img_floats.size() * sizeof(float)); in BuildImageTensor()
192 std::memcpy(img_tensor.flat<float>().data(), img_floats.data(), in BuildImageTensor()
193 img_tensor.TotalBytes()); in BuildImageTensor()
194 return img_tensor; in BuildImageTensor()
200 Tensor img_tensor(DT_QUINT8, {1, WIDTH, HEIGHT, DEPTH}); in BuildQuantizedImageTensor() local
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()
205 img_tensor.TotalBytes()); in BuildQuantizedImageTensor()
206 return img_tensor; in BuildQuantizedImageTensor()
[all …]
|