|
|
|
@ -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];
|
|
|
|
|