Searched refs:image_quantized_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_resize_bilinear_op_test.cc | 241 Tensor image_quantized_tensor(DT_QINT32, TensorShape({1, 1, IN_WIDTH, 1})); in TestResizeBilinearOneDim() local
243 for (int64 i = 0; i < image_quantized_tensor.NumElements(); ++i) { in TestResizeBilinearOneDim()
244 image_quantized_tensor.flat<qint32>()(i) = in TestResizeBilinearOneDim()
249 TestResizeBilinear(image_quantized_tensor, DT_QINT32, {1, OUT_WIDTH}, false, in TestResizeBilinearOneDim()
263 image_quantized_tensor.flat<qint32>()(i / 2), MIN, MAX); in TestResizeBilinearOneDim()
266 image_quantized_tensor.flat<qint32>()(i / 2), MIN, MAX); in TestResizeBilinearOneDim()
268 image_quantized_tensor.flat<qint32>()(i / 2 + 1), MIN, MAX); in TestResizeBilinearOneDim()
277 CheckTensorValue<qint32>(image_quantized_tensor.flat<qint32>().data(), in TestResizeBilinearOneDim()
296 const Tensor image_quantized_tensor = BuildTensor<T>( in RunTestResizeBilinearTwoDims() local
300 TestResizeBilinear(image_quantized_tensor, DataTypeToEnum<T>::value, in RunTestResizeBilinearTwoDims()
[all …]
|