Public Member Functions | |
NetMutator (caffe2::NetDef *net) | |
NetMutator & | newOp (const std::string &type, const std::vector< std::string > &inputs, const std::vector< std::string > &outputs) |
NetMutator & | externalInputs (const std::vector< std::string > &externalInputs) |
NetMutator & | externalOutputs (const std::vector< std::string > &externalOutputs) |
template<typename T > | |
NetMutator & | addArgument (const std::string &name, const T &value) |
NetMutator & | setDeviceOptionName (const std::string &name) |
Definition at line 127 of file test_utils.h.