pytorch  1.8.2
About: PyTorch provides Tensor computation (like NumPy) with strong GPU acceleration and Deep Neural Networks (in Python) built on a tape-based autograd system. LTS (Long Term Support) release.
  Fossies Dox: pytorch-1.8.2.tar.gz  ("unofficial" and yet experimental doxygen-generated source code documentation)  

caffe2::BooleanMaskOpGradient< Context > Class Template Referencefinal

#include <boolean_mask_ops.h>

Inheritance diagram for caffe2::BooleanMaskOpGradient< Context >:
[legend]
Collaboration diagram for caffe2::BooleanMaskOpGradient< Context >:
[legend]

Public Member Functions

 BooleanMaskOpGradient (const OperatorDef &operator_def, Workspace *ws)
 
bool RunOnDevice () override
 
template<typename T >
bool DoRunWithType ()
 
bool DoRunWithType ()
 
- Public Member Functions inherited from caffe2::Operator< Context >
 Operator (const OperatorDef &operator_def, Workspace *ws)
 
 Operator (const c10::FunctionSchema &fn_schema, std::vector< c10::IValue > inputs, c10::List< at::Tensor > outputs)
 
 ~Operator () noexcept override
 
const TensorInput (int idx, DeviceType type=Context::GetDeviceType())
 Retrieve a non-owning reference to the input at position 'idx' for this operator. More...
 
Tensor XOutput (int idx, at::IntArrayRef dims, at::TensorOptions options)
 XOutput is a modernized version of Output which returns a Tensor rather than a Tensor* (the raw pointer in the latter case is useless, as Tensor is a pointer type.) More...
 
- Public Member Functions inherited from caffe2::OperatorBase
 OperatorBase (const OperatorDef &operator_def, Workspace *ws)
 
 OperatorBase (const c10::FunctionSchema &schema, std::vector< c10::IValue > inputs, c10::List< at::Tensor > outputs)
 
virtual ~OperatorBase () noexcept
 
bool isLegacyOperator () const
 Return true if the operator was instantiated with OperatorDef New operators should be instantiated with FunctionSchema. More...
 
const c10::FunctionSchemagetFunctionSchema () const
 
bool HasArgument (const string &name) const
 Checks if the operator has an argument of the given name. More...
 
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< TGetVectorFromIValueList (const c10::IValue &value) const
 
template<typename T >
vector< TGetRepeatedArgument (const string &name, const vector< T > &default_value={}) const
 
template<typename T >
const TInput (int idx)
 
template<typename T >
const TInput (int idx, DeviceType type)
 
template<typename T >
TOutput (int idx)
 
template<typename T >
TOutput (int idx, DeviceType type)
 
Tensor XOutputTensor (int idx, at::IntArrayRef dims, at::TensorOptions options)
 
void SetOutputTensor (int idx, Tensor tensor)
 
Tensor OutputTensorOrUndefined (int idx)
 
TensorOutputTensor (int idx, at::IntArrayRef dims, at::TensorOptions options)
 
TensorOutputTensorCopyFrom (int idx, at::TensorOptions options, const Tensor &src, bool async=false)
 
TensorOutputTensorAlias (int idx, const Tensor &src)
 
template<typename T >
TOutput (int idx, T *allocated)
 
const BlobInputBlob (int idx)
 
BlobOutputBlob (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
 
virtual void WaitEvent (const Event &ev, int=-1)
 
void Wait (const OperatorBase &other, int stream_id=-1)
 
virtual void WaitEvents (const std::vector< const Event * > &events, int=-1)
 
virtual void Finish ()
 
virtual bool Run (int=0)
 
virtual bool HasAsyncPart () const
 
virtual bool SupportsAsyncScheduling () const
 
virtual void CancelAsyncCallback ()
 
virtual void Cancel ()
 
virtual bool RunAsync (int stream_id=0)
 
virtual void AddRelatedBlobInfo (EnforceNotMet *err)
 
virtual std::string debug_info_string () const
 
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 Eventevent () const
 
Eventevent ()
 
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)
 
ExecutorHelperGetExecutorHelper () const
 
c10::List< at::Tensormove_newstyle_outputs () &&
 
template<>
NetDef GetSingleArgument (const std::string &name, const NetDef &default_value) const
 
template<>
vector< intGetVectorFromIValueList (const c10::IValue &value) const
 
template<>
vector< floatGetVectorFromIValueList (const c10::IValue &value) const
 
template<>
vector< string > GetVectorFromIValueList (const c10::IValue &value) const
 
template<>
vector< int16_t > GetVectorFromIValueList (const c10::IValue &value) const
 
template<>
vector< int16_t > GetRepeatedArgument (const string &name, const vector< int16_t > &default_value) const
 
- Public Member Functions inherited from caffe2::Observable< OperatorBase >
 Observable ()=default
 
 Observable (Observable &&)=default
 
Observableoperator= (Observable &&)=default
 
virtual ~Observable ()=default
 
 C10_DISABLE_COPY_AND_ASSIGN (Observable)
 
const ObserverAttachObserver (std::unique_ptr< Observer > observer)
 
std::unique_ptr< ObserverDetachObserver (const Observer *observer_ptr)
 Returns a unique_ptr to the removed observer. More...
 
virtual size_t NumObservers ()
 
void StartAllObservers ()
 
void StopAllObservers ()
 

Public Attributes

 USE_OPERATOR_CONTEXT_FUNCTIONS
 

Additional Inherited Members

- Public Types inherited from caffe2::Observable< OperatorBase >
using Observer = ObserverBase< OperatorBase >
 
- Static Public Attributes inherited from caffe2::OperatorBase
static const int kNoNetPositionSet = -1
 
- Protected Member Functions inherited from caffe2::OperatorBase
virtual void RecordEvent (const char *=nullptr)
 
void SetEventFinished (const char *err_msg=nullptr)
 
void SetEventFinishedWithException (const char *err_msg=nullptr)
 
std::string getErrorMsg ()
 
c10::optional< intargumentIndexWithName (const std::string &name) const
 
 C10_DISABLE_COPY_AND_ASSIGN (OperatorBase)
 
- Protected Attributes inherited from caffe2::OperatorBase
std::unique_ptr< Eventevent_
 
- Protected Attributes inherited from caffe2::Observable< OperatorBase >
std::vector< std::unique_ptr< Observer > > observers_list_
 

Detailed Description

template<class Context>
class caffe2::BooleanMaskOpGradient< Context >

Definition at line 23 of file boolean_mask_ops.h.

Constructor & Destructor Documentation

◆ BooleanMaskOpGradient()

template<class Context >
caffe2::BooleanMaskOpGradient< Context >::BooleanMaskOpGradient ( const OperatorDef &  operator_def,
Workspace ws 
)
inline

Definition at line 26 of file boolean_mask_ops.h.

Member Function Documentation

◆ DoRunWithType() [1/2]

◆ DoRunWithType() [2/2]

template<class Context >
template<typename T >
bool caffe2::BooleanMaskOpGradient< Context >::DoRunWithType ( )

◆ RunOnDevice()

template<class Context >
bool caffe2::BooleanMaskOpGradient< Context >::RunOnDevice ( )
inlineoverride

Definition at line 33 of file boolean_mask_ops.h.

References call(), and caffe2::Operator< Context >::Input().

Member Data Documentation

◆ USE_OPERATOR_CONTEXT_FUNCTIONS

template<class Context >
caffe2::BooleanMaskOpGradient< Context >::USE_OPERATOR_CONTEXT_FUNCTIONS

Definition at line 25 of file boolean_mask_ops.h.


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