| 445 | } |
| 446 | |
| 447 | bool GraphExecutor::SetTensorData(Tensor* tensor, const void* input_data, int data_size) |
| 448 | { |
| 449 | int tensor_size = tensor->GetTotalSize(); |
| 450 | |
| 451 | if(tensor_size != data_size) |
| 452 | return false; |
| 453 | |
| 454 | void* tensor_addr = GetTensorBuffer(tensor); |
| 455 | |
| 456 | if(tensor_addr == nullptr) |
| 457 | return false; |
| 458 | |
| 459 | std::memcpy(tensor_addr, input_data, data_size); |
| 460 | |
| 461 | return true; |
| 462 | } |
| 463 | |
| 464 | bool GraphExecutor::GetTensorData(Tensor* tensor, void* output_data, int data_size) |
| 465 | { |
nothing calls this directly
no test coverage detected