|
|
|
@@ -38,7 +38,7 @@ |
|
|
|
#include "parser/tensorflow/tensorflow_ref_switch_parser.h" |
|
|
|
#include "parser/tensorflow/tensorflow_fusion_op_parser.h" |
|
|
|
#include "parser/tensorflow/tensorflow_auto_mapping_parser_adapter.h" |
|
|
|
#include "parser/common/op_def/arg_op.h" |
|
|
|
#include "parser/common/op_def/arg_op_operator.h" |
|
|
|
#include "parser/tensorflow/tensorflow_fusion_custom_parser_adapter.h" |
|
|
|
#include "parser/tensorflow/tensorflow_reshape_parser.h" |
|
|
|
#include "parser/tensorflow/tensorflow_custom_parser_adapter.h" |
|
|
|
@@ -678,6 +678,7 @@ namespace { |
|
|
|
|
|
|
|
if ((_name== "S") || (_name == "K")) { |
|
|
|
int index = 0; |
|
|
|
|
|
|
|
ge::AttrUtils::SetInt(opDef, "T", 1); |
|
|
|
ge::AttrUtils::SetInt(opDef, "arg_index", index); |
|
|
|
ge::AttrUtils::SetInt(opDef, "ret_index", index); |
|
|
|
|