Browse Source

update corresponding header files

tags/v0.7.0-beta
zhangzhenghai 4 years ago
parent
commit
9275e7b08e
4 changed files with 4 additions and 38 deletions
  1. +4
    -11
      inc/graph/utils/graph_utils.h
  2. +0
    -25
      inc/graph/utils/node_utils.h
  3. +0
    -1
      inc/graph/utils/tensor_adapter.h
  4. +0
    -1
      inc/graph/utils/tensor_utils.h

+ 4
- 11
inc/graph/utils/graph_utils.h View File

@@ -130,7 +130,7 @@ struct NodeIndexIO {
IOType io_type_ = kOut;
std::string value_;

const std::string &ToString() const { return value_; }
std::string ToString() const { return value_; }
};

class GraphUtils {
@@ -188,8 +188,8 @@ class GraphUtils {
/// @param [in] output_index
/// @return graphStatus
///
static graphStatus InsertNodeAfter(const OutDataAnchorPtr &src, const std::vector<InDataAnchorPtr> &dsts,
const NodePtr &insert_node, uint32_t input_index = 0, uint32_t output_index = 0);
static graphStatus InsertNodeBefore(const OutDataAnchorPtr &src, const std::vector<InDataAnchorPtr> &dsts,
const NodePtr &insert_node, uint32_t input_index = 0, uint32_t output_index = 0);

static graphStatus RemoveJustNode(ComputeGraphPtr compute_graph, const NodePtr &node);

@@ -303,14 +303,6 @@ class GraphUtils {
///
static graphStatus MoveOutCtrlEdges(NodePtr &src_node, NodePtr &dst_node);

///
/// Copy all in-data edges from `src_node` to `dst_node`
/// @param src_node
/// @param dst_node
/// @return
///
static graphStatus CopyInDataEdges(const NodePtr &src_node, NodePtr &dst_node);

static ComputeGraphPtr FindRootGraph(ComputeGraphPtr graph);

static graphStatus TopologicalSortingByName(const ge::ComputeGraphPtr &compute_graph, vector<NodePtr> &node_vec);
@@ -736,4 +728,5 @@ class PartialGraphBuilder : public ComputeGraphBuilder {
std::vector<NodePtr> exist_nodes_;
};
} // namespace ge

#endif // INC_GRAPH_UTILS_GRAPH_UTILS_H_

+ 0
- 25
inc/graph/utils/node_utils.h View File

@@ -99,13 +99,6 @@ class NodeUtils {
///
static NodePtr GetParentInput(const NodePtr &node);

///
/// @brief Check is varying_input for while node
/// @param [in] node: Data node for subgraph
/// @return bool
///
static bool IsWhileVaryingInput(const ge::NodePtr &node);

///
/// @brief Get subgraph input is constant.
/// @param [in] node
@@ -121,24 +114,6 @@ class NodeUtils {
///
static graphStatus RemoveSubgraphsOnNode(const NodePtr &node);

///
/// @brief Get subgraph input data node by index.
/// @param [in] node
/// @return Node
///
static vector<NodePtr> GetSubgraphDataNodesByIndex(const Node &node, int index);

///
/// @brief Get subgraph input data node by index.
/// @param [in] node
/// @return Node
///
static vector<NodePtr> GetSubgraphOutputNodes(const Node &node);

static NodePtr GetInDataNodeByIndex(const Node &node, int index);

static vector<NodePtr> GetOutDataNodesByIndex(const Node &node, int index);

private:
static std::map<NodePtr, std::vector<uint32_t>> map_send_info_;
static std::map<NodePtr, std::vector<uint32_t>> map_recv_info_;


+ 0
- 1
inc/graph/utils/tensor_adapter.h View File

@@ -20,7 +20,6 @@
#include <memory>
#include "graph/ge_tensor.h"
#include "graph/tensor.h"

namespace ge {
using GeTensorPtr = std::shared_ptr<GeTensor>;
using ConstGeTensorPtr = std::shared_ptr<const GeTensor>;


+ 0
- 1
inc/graph/utils/tensor_utils.h View File

@@ -21,7 +21,6 @@
#include "graph/def_types.h"
#include "graph/ge_error_codes.h"
#include "graph/ge_tensor.h"

namespace ge {
class TensorUtils {
public:


Loading…
Cancel
Save