|
|
|
@@ -120,7 +120,8 @@ bool AkgKernelJsonGenerator::CreateInputDescJson(const AnfNodePtr &anf_node, con |
|
|
|
input_desc_json[kJsonKeyName] = op_input_name; |
|
|
|
input_desc_json[kJsonKeyTensorName] = "input_" + std::to_string(GetInputTensorIdxInc(anf_node, real_input_index)); |
|
|
|
auto input_shape = this->GetInputShape(anf_node, real_input_index); |
|
|
|
if (GetInputTensorValue(anf_node, real_input_index, &input_desc_json)) { |
|
|
|
if (anf_node->func_graph() != nullptr && anf_node->func_graph()->has_attr(FUNC_GRAPH_ATTR_GRAPH_KERNEL) && |
|
|
|
GetInputTensorValue(anf_node, real_input_index, &input_desc_json)) { |
|
|
|
MS_LOG(DEBUG) << "Take input[" << real_input_index << "] of [" << anf_node->DebugString(2) |
|
|
|
<< "] as const tensor, shape: [" << Vector2Str(input_shape) |
|
|
|
<< "], value: " << input_desc_json[kJsonKeyValue]; |
|
|
|
|