mxnet
Public Member Functions | List of all members
mxnet::ext::CustomOpSelector Class Referenceabstract

#include <lib_api.h>

Collaboration diagram for mxnet::ext::CustomOpSelector:
Collaboration graph

Public Member Functions

virtual bool Select (int nodeID)=0
 
virtual bool SelectInput (int nodeID, int input_nodeID)=0
 
virtual bool SelectOutput (int nodeID, int output_nodeID)=0
 
virtual void Filter (const std::vector< int > &candidates, std::vector< int > *keep)
 
virtual void Reset ()
 

Member Function Documentation

◆ Filter()

virtual void mxnet::ext::CustomOpSelector::Filter ( const std::vector< int > &  candidates,
std::vector< int > *  keep 
)
inlinevirtual

◆ Reset()

virtual void mxnet::ext::CustomOpSelector::Reset ( )
inlinevirtual

◆ Select()

virtual bool mxnet::ext::CustomOpSelector::Select ( int  nodeID)
pure virtual

◆ SelectInput()

virtual bool mxnet::ext::CustomOpSelector::SelectInput ( int  nodeID,
int  input_nodeID 
)
pure virtual

◆ SelectOutput()

virtual bool mxnet::ext::CustomOpSelector::SelectOutput ( int  nodeID,
int  output_nodeID 
)
pure virtual

The documentation for this class was generated from the following file: