|
|
@ -424,7 +424,7 @@ Status GraphBuilder::GetTaskInfo(const ge::ModelBuilder &builder, const ModelPtr
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Status GraphBuilder::SetInputSize(const ge::NodePtr &node_ptr) {
|
|
|
|
Status GraphBuilder::SetInputSize(const ge::NodePtr &node_ptr) {
|
|
|
|
// set input_desc.size = src_node.output_desc.size
|
|
|
|
// Set the size of input_desc to 'src_node.output_desc.size'
|
|
|
|
if (node_ptr->GetType() == DATA) {
|
|
|
|
if (node_ptr->GetType() == DATA) {
|
|
|
|
bool is_unknown_shape = false;
|
|
|
|
bool is_unknown_shape = false;
|
|
|
|
GE_CHK_STATUS_RET(ge::NodeUtils::GetNodeUnknownShapeStatus(*node_ptr, is_unknown_shape),
|
|
|
|
GE_CHK_STATUS_RET(ge::NodeUtils::GetNodeUnknownShapeStatus(*node_ptr, is_unknown_shape),
|
|
|
@ -447,7 +447,7 @@ Status GraphBuilder::SetInputSize(const ge::NodePtr &node_ptr) {
|
|
|
|
GE_IF_BOOL_EXEC(src_op == nullptr, continue);
|
|
|
|
GE_IF_BOOL_EXEC(src_op == nullptr, continue);
|
|
|
|
auto node_op_desc = node_ptr->GetOpDesc();
|
|
|
|
auto node_op_desc = node_ptr->GetOpDesc();
|
|
|
|
GE_IF_BOOL_EXEC(node_op_desc == nullptr, continue);
|
|
|
|
GE_IF_BOOL_EXEC(node_op_desc == nullptr, continue);
|
|
|
|
// set dst_node.input_desc = src_node.output_desc
|
|
|
|
// Set the input_desc of dst_node to 'src_node.output_desc'
|
|
|
|
auto output_desc = src_op->GetOutputDescPtr(peer_out_anchor->GetIdx());
|
|
|
|
auto output_desc = src_op->GetOutputDescPtr(peer_out_anchor->GetIdx());
|
|
|
|
int64_t size = 0;
|
|
|
|
int64_t size = 0;
|
|
|
|
GE_IF_BOOL_EXEC(ge::TensorUtils::GetSize(*output_desc, size) != SUCCESS, GELOGI("Get size failed!"));
|
|
|
|
GE_IF_BOOL_EXEC(ge::TensorUtils::GetSize(*output_desc, size) != SUCCESS, GELOGI("Get size failed!"));
|
|
|
|