|
| USE_IDEEP_DEF_ALIASES () |
|
| USE_IDEEP_OPERATOR_FUNCTIONS () |
|
| IDEEPMomentumSGDOp (const OperatorDef &operator_def, Workspace *ws) |
|
bool | RunOnDevice () override |
|
| IDEEPOperator (const OperatorDef &operator_def, Workspace *ws) |
|
const ideep::tensor & | Input (int index) |
|
ideep::tensor * | Output (int index) |
|
bool | Run (int) final |
|
void | WaitEvent (const Event &ev, int) final |
|
void | WaitEvents (const std::vector< const Event * > &events, int) final |
|
void | RecordEvent (const char *err_msg=nullptr) final |
|
| OperatorBase (const OperatorDef &operator_def, Workspace *ws) |
|
| OperatorBase (const c10::FunctionSchema &schema, std::vector< c10::IValue > inputs, std::vector< at::Tensor > outputs) |
|
bool | isLegacyOperator () const |
| Return true if the operator was instantiated with OperatorDef New operators should be instantiated with FunctionSchema.
|
|
const c10::FunctionSchema & | getFunctionSchema () const |
|
bool | HasArgument (const string &name) const |
| Checks if the operator has an argument of the given name.
|
|
template<typename T > |
T | GetSingleArgument (const string &name, const T &default_value) const |
|
template<typename T > |
bool | HasSingleArgumentOfType (const string &name) const |
|
template<typename T > |
vector< T > | GetVectorFromIValueList (const c10::IValue &value) const |
|
template<typename T > |
vector< T > | GetRepeatedArgument (const string &name, const vector< T > &default_value={}) const |
|
template<typename T > |
const T & | Input (int idx) |
|
template<typename T > |
const T & | Input (int idx, DeviceType type) |
|
template<typename T > |
T * | Output (int idx) |
|
template<typename T > |
T * | Output (int idx, DeviceType type) |
|
Tensor | XOutputTensor (int idx, at::IntArrayRef dims, at::TensorOptions options) |
|
void | SetOutputTensor (int idx, Tensor tensor) |
|
Tensor | OutputTensorOrUndefined (int idx) |
|
Tensor * | OutputTensor (int idx, at::IntArrayRef dims, at::TensorOptions options) |
|
Tensor * | OutputTensorCopyFrom (int idx, at::TensorOptions options, const Tensor &src, bool async=false) |
|
Tensor * | OutputTensorAlias (int idx, const Tensor &src) |
|
template<typename T > |
T * | Output (int idx, T *allocated) |
|
const Blob & | InputBlob (int idx) |
|
Blob * | OutputBlob (int idx) |
|
bool | IsInputOutputAlias (int i, int j) |
|
template<typename T > |
bool | InputIsType (int idx) |
|
bool | InputIsTensorType (int idx, DeviceType device_type) |
|
template<typename T > |
bool | OutputIsType (int idx) |
|
bool | OutputIsTensorType (int idx, DeviceType type) |
|
int | InputSize () const |
|
int | OutputSize () const |
|
const vector< const Blob * > & | Inputs () const |
|
const vector< Blob * > & | Outputs () |
|
vector< TensorShape > | InputTensorShapes () const |
|
void | Wait (const OperatorBase &other, int stream_id=-1) |
|
virtual void | Finish () |
|
virtual bool | HasAsyncPart () const |
|
virtual bool | SupportsAsyncScheduling () const |
|
virtual bool | RunAsync (int stream_id=0) |
|
virtual void | AddRelatedBlobInfo (EnforceNotMet *err) |
|
const OperatorDef & | debug_def () const |
|
void | set_debug_def (const std::shared_ptr< const OperatorDef > &operator_def) |
|
bool | has_debug_def () const |
|
void | RecordLastFailedOpNetPosition () |
|
int | net_position () const |
|
void | set_net_position (int idx) |
|
const DeviceOption & | device_option () const |
|
const Event & | event () const |
|
Event & | event () |
|
void | ResetEvent () |
|
void | DisableEvent () |
|
bool | IsEventDisabled () const |
|
virtual void | SyncDeviceBarrierForObservers () |
|
virtual bool | IsStreamFree (int) const |
|
const std::string & | type () const |
|
void | annotate_engine (const std::string &engine) |
|
const std::string & | engine () const |
|
void | SetExecutorHelper (ExecutorHelper *helper) |
|
ExecutorHelper * | GetExecutorHelper () const |
|
std::vector< at::Tensor > | move_newstyle_outputs ()&& |
|
template<> |
NetDef | GetSingleArgument (const std::string &name, const NetDef &default_value) const |
|
template<> |
vector< int > | GetVectorFromIValueList (const c10::IValue &value) const |
|
template<> |
vector< float > | GetVectorFromIValueList (const c10::IValue &value) const |
|
template<> |
vector< string > | GetVectorFromIValueList (const c10::IValue &value) const |
|
| Observable (Observable &&)=default |
|
Observable & | operator= (Observable &&)=default |
|
| C10_DISABLE_COPY_AND_ASSIGN (Observable) |
|
const Observer * | AttachObserver (std::unique_ptr< Observer > observer) |
|
std::unique_ptr< Observer > | DetachObserver (const Observer *observer_ptr) |
| Returns a unique_ptr to the removed observer. More...
|
|
virtual size_t | NumObservers () |
|
void | StartAllObservers () |
|
void | StopAllObservers () |
|
Definition at line 37 of file momentum_sgd_op.cc.