op_desc.h 3.6 KB
Newer Older
F
fengjiayi 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */

#pragma once

#include <unordered_map>
#include <vector>
#include "paddle/framework/attribute.h"
20
#include "paddle/framework/type_defs.h"
F
fengjiayi 已提交
21 22 23 24 25 26 27 28 29
#include "paddle/framework/var_desc.h"

namespace paddle {
namespace framework {

class BlockDescBind;

class OpDescBind {
 public:
F
fengjiayi 已提交
30 31 32 33 34
  OpDescBind() {}

  OpDescBind(const std::string &type, const VariableNameMap &inputs,
             const VariableNameMap &outputs, const AttributeMap &attrs);

F
fengjiayi 已提交
35 36 37 38 39 40 41 42
  OpDesc *Proto();

  std::string Type() const { return op_desc_.type(); }

  void SetType(const std::string &type) { op_desc_.set_type(type); }

  const std::vector<std::string> &Input(const std::string &name) const;

F
Update  
fengjiayi 已提交
43 44
  std::vector<std::string> InputArgumentNames() const;

F
fengjiayi 已提交
45 46 47 48 49
  void SetInput(const std::string &param_name,
                const std::vector<std::string> &args);

  const std::vector<std::string> &Output(const std::string &name) const;

F
Update  
fengjiayi 已提交
50 51
  std::vector<std::string> OutputArgumentNames() const;

F
fengjiayi 已提交
52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
  void SetOutput(const std::string &param_name,
                 const std::vector<std::string> &args);

  bool HasAttr(const std::string &name) const {
    return attrs_.find(name) != attrs_.end();
  }

  AttrType GetAttrType(const std::string &name) const;

  std::vector<std::string> AttrNames() const;

  void SetAttr(const std::string &name, const Attribute &v);

  void SetBlockAttr(const std::string &name, BlockDescBind &block);

  Attribute GetAttr(const std::string &name) const;

  int GetBlockAttr(const std::string &name) const;

F
fengjiayi 已提交
71 72
  void Rename(const std::string &old_name, const std::string &new_name);

F
fengjiayi 已提交
73
  // Only be used in C++
Y
Yu Yang 已提交
74
  const AttributeMap &GetAttrMap() const;
F
fengjiayi 已提交
75

76 77
  // Only be used in C++
  void SetAttrMap(const AttributeMap &attr_map);
78

Y
Yu Yang 已提交
79 80
  std::vector<std::string> InputNames() const { return MapKeys(inputs_); }
  std::vector<std::string> OutputNames() const { return MapKeys(outputs_); }
81

Y
Yu Yang 已提交
82
  void SetInputMap(const VariableNameMap &input) {
Y
Stash  
Yu Yang 已提交
83 84 85 86
    this->inputs_ = input;
    this->need_update_ = true;
  }

Y
Yu Yang 已提交
87
  void SetOutputMap(const VariableNameMap &output) {
Y
Stash  
Yu Yang 已提交
88 89 90 91
    this->outputs_ = output;
    this->need_update_ = true;
  }

Y
Yu Yang 已提交
92 93 94 95 96 97
  void Sync();

  const VariableNameMap &Inputs() const { return inputs_; }

  const VariableNameMap &Outputs() const { return outputs_; }

98 99 100 101 102
  AttributeMap *MutableAttrMap() {
    this->need_update_ = true;
    return &this->attrs_;
  }

Y
Yu Yang 已提交
103 104
  void InferShape(const BlockDescBind &block) const;

F
fengjiayi 已提交
105
 private:
106 107 108 109 110
  template <typename MapType>
  static std::vector<typename MapType::key_type> MapKeys(const MapType &map) {
    std::vector<typename MapType::key_type> ret_val;
    ret_val.reserve(map.size());
    std::transform(
Y
Fix CI  
Yu Yang 已提交
111
        map.begin(), map.end(), std::back_inserter(ret_val),
112 113 114 115
        [](const typename MapType::value_type &pair) { return pair.first; });
    return ret_val;
  }

F
fengjiayi 已提交
116
  OpDesc op_desc_;
Y
Yu Yang 已提交
117 118 119
  VariableNameMap inputs_;
  VariableNameMap outputs_;
  AttributeMap attrs_;
F
fengjiayi 已提交
120 121 122 123 124

  // need_update_ indicate there some local changes not be synchronized. If
  // local changes should be synchronized, need_update_ should be set to true.
  bool need_update_{false};
};
F
fengjiayi 已提交
125 126
}  // namespace framework
}  // namespace paddle