| 431 | } |
| 432 | |
| 433 | tensorflow::Status ImportInt64Array(const TensorProto& input_tensor, |
| 434 | Array* output_array) { |
| 435 | CHECK_EQ(input_tensor.dtype(), DT_INT64); |
| 436 | const auto& input_shape = input_tensor.tensor_shape(); |
| 437 | CHECK_LE(input_shape.dim_size(), 6); |
| 438 | int input_flat_size; |
| 439 | auto status = ImportShape(input_shape.dim(), &input_flat_size, |
| 440 | output_array->mutable_shape()); |
| 441 | if (!status.ok()) return status; |
| 442 | |
| 443 | auto& output_int_data = |
| 444 | output_array->GetMutableBuffer<ArrayDataType::kInt64>().data; |
| 445 | output_int_data.resize(RequiredBufferSizeForShape(output_array->shape()), 0); |
| 446 | return ImportTensorData<int64>(input_tensor, input_flat_size, |
| 447 | &output_int_data); |
| 448 | } |
| 449 | |
| 450 | tensorflow::Status ImportBoolArray(const TensorProto& input_tensor, |
| 451 | Array* output_array) { |
no test coverage detected