| 1459 | } |
| 1460 | |
| 1461 | void IDeserializer::DeserializerImpl::ParseCast(GraphPtr graph, unsigned int layerIndex) |
| 1462 | { |
| 1463 | CHECK_LAYERS(graph, 0, layerIndex); |
| 1464 | TensorRawPtrVector inputs = GetInputs(graph, layerIndex); |
| 1465 | CHECK_LOCATION(); |
| 1466 | CHECK_VALID_SIZE(inputs.size(), 1); |
| 1467 | |
| 1468 | TensorRawPtrVector outputs = GetOutputs(graph, layerIndex); |
| 1469 | CHECK_VALID_SIZE(outputs.size(), 1); |
| 1470 | |
| 1471 | auto layerName = GetLayerName(graph, layerIndex); |
| 1472 | |
| 1473 | IConnectableLayer* layer = m_Network->AddCastLayer(layerName.c_str()); |
| 1474 | |
| 1475 | armnn::TensorInfo outputTensorInfo = ToTensorInfo(outputs[0]); |
| 1476 | layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); |
| 1477 | |
| 1478 | RegisterInputSlots(graph, layerIndex, layer); |
| 1479 | RegisterOutputSlots(graph, layerIndex, layer); |
| 1480 | } |
| 1481 | |
| 1482 | void IDeserializer::DeserializerImpl::ParseConstant(GraphPtr graph, unsigned int layerIndex) |
| 1483 | { |
nothing calls this directly
no test coverage detected