| 397 | } |
| 398 | |
| 399 | tensorflow::Status ImportQuint8Array(const TensorProto& input_tensor, |
| 400 | Array* output_array) { |
| 401 | CHECK_EQ(input_tensor.dtype(), DT_QUINT8); |
| 402 | const auto& input_shape = input_tensor.tensor_shape(); |
| 403 | CHECK_LE(input_shape.dim_size(), 6); |
| 404 | int input_flat_size; |
| 405 | auto status = ImportShape(input_shape.dim(), &input_flat_size, |
| 406 | output_array->mutable_shape()); |
| 407 | if (!status.ok()) return status; |
| 408 | |
| 409 | auto& output_int_data = |
| 410 | output_array->GetMutableBuffer<ArrayDataType::kUint8>().data; |
| 411 | output_int_data.resize(RequiredBufferSizeForShape(output_array->shape()), 0); |
| 412 | return ImportTensorData<uint8_t>(input_tensor, input_flat_size, |
| 413 | &output_int_data); |
| 414 | } |
| 415 | |
| 416 | tensorflow::Status ImportInt32Array(const TensorProto& input_tensor, |
| 417 | Array* output_array) { |
no test coverage detected