Crate caffe2_graph

source ·

Structs

  • | Graph representation of a Netdef. |
  • | Graph representation of an operator. | | Default constructor for resize |

Traits

Functions

  • | Adds an operator def to a netdef. | | Returns the ptr, if you want to add anything | extra (such as device_option)
  • | Checks if two netdefs are in terms of | type, input, and output. |
  • | This ensures that each named arg that | exists in the pattern exists in g_op, | is equal in value. |
  • | This allows for the use of * and | to match | operator types, engines, or any other | property that is represented by strings. | | For example, if we wanted to match an | operator to Conv or FC, we can give: “Conv|FC” | as the type() of that op. |