|
|
@@ -34,17 +34,17 @@ |
|
|
|
#include "external/parser/tensorflow_parser.h" |
|
|
|
#include "parser/tensorflow/tensorflow_constant_parser.h" |
|
|
|
#include "common/types.h" |
|
|
|
#include "parser/common/op_def/variable_op.h" |
|
|
|
#include "parser/common/op_def/variable_operator.h" |
|
|
|
#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" |
|
|
|
#include "parser/tensorflow/tensorflow_squeeze_parser.h" |
|
|
|
#include "parser/tensorflow/graph_functiondef.h" |
|
|
|
#include "parser/tensorflow/graph_optimizer.h" |
|
|
|
#include "parser/tensorflow/graph_to_function_def.h" |
|
|
|
#include "parser/tensorflow/parser_graph_optimizer.h" |
|
|
|
#include "cce/dnn_base_def.hpp" |
|
|
|
#include "parser/tensorflow/scope/scope_pass_manager.h" |
|
|
|
#include "parser/tensorflow/tensorflow_util.h" |
|
|
@@ -52,10 +52,10 @@ |
|
|
|
#include "parser/tensorflow/tensorflow_enter_parser.h" |
|
|
|
#include "parser/common/op_def/ir_pb_converter.h" |
|
|
|
#include "parser/common/tuple.h" |
|
|
|
#include "common/op_def/frameworkop_op.h" |
|
|
|
#include "common/op_def/shape_n_op.h" |
|
|
|
#include "common/op_def/var_is_initialized_op_op.h" |
|
|
|
#include "common/op_def/fill_op.h" |
|
|
|
#include "common/op_def/framework_op_operator.h" |
|
|
|
#include "common/op_def/shape_n_operator.h" |
|
|
|
#include "common/op_def/var_is_initialized_op_operator.h" |
|
|
|
#include "common/op_def/fill_operator.h" |
|
|
|
#include "common/convert/pb2json.h" |
|
|
|
#include "common/convert/message2operator.h" |
|
|
|
#include "parser/common/proto_file_parser.h" |
|
|
@@ -70,7 +70,7 @@ |
|
|
|
#include "parser/common/prototype_pass_manager.h" |
|
|
|
#include "parser/common/register_tbe.h" |
|
|
|
#include "parser/common/pass_manager.h" |
|
|
|
#include "parser/tensorflow/graph_optimizer.h" |
|
|
|
#include "parser/tensorflow/parser_graph_optimizer.h" |
|
|
|
#include "metadef/inc/register/scope/scope_pass_registry_impl.h" |
|
|
|
#include "register/scope/scope_fusion_pass_register.h" |
|
|
|
#undef protected |
|
|
@@ -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); |
|
|
|