ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Types | Public Member Functions
onert::ir::operation::Loss Class Reference

#include <Loss.h>

Collaboration diagram for onert::ir::operation::Loss:
Collaboration graph
[legend]

Public Types

enum  Input { Y_PRED = 0 , Y_TRUE = 1 }
 

Public Member Functions

 Loss (const OperandIndexSequence &inputs, const OperandIndexSequence &outputs)
 
void accept (OperationVisitor &v) const override
 
OpCode opcode () const final
 
- Public Member Functions inherited from onert::ir::Operation
 Operation (OperandConstraint input_constr, const OperandIndexSequence &inputs, const OperandIndexSequence &outputs, OperandConstraint output_constr=OperandConstraint::createAny())
 
 Operation (OperandConstraint input_constr, OperandConstraint output_constr=OperandConstraint::createAny())
 
 Operation (const Operation &)=default
 
 Operation (Operation &&)=default
 
Operation & operator= (const Operation &)=default
 
Operation & operator= (Operation &&)=default
 
virtual ~Operation ()
 
void replaceInputs (const OperandIndex &from, const OperandIndex &to) override
 
void replaceOutputs (const OperandIndex &from, const OperandIndex &to) override
 
OperandIndexSequence & getInputs ()
 
const OperandIndexSequence & getInputs () const override
 
const OperandIndexSequence & getOutputs () const override
 
void setInputs (const OperandIndexSequence &indexes)
 
void setOutputs (const OperandIndexSequence &indexes)
 
- Public Member Functions inherited from onert::ir::IOperation
virtual ~IOperation ()=default
 
virtual std::string name () const
 

Detailed Description

Definition at line 29 of file Loss.h.

Member Enumeration Documentation

◆ Input

enum onert::ir::operation::Loss::Input
Enumerator
Y_PRED 
Y_TRUE 

Definition at line 32 of file Loss.h.

33 {
34 Y_PRED = 0,
35 Y_TRUE = 1
36 // TODO Add more inputs if necessary
37 };

Constructor & Destructor Documentation

◆ Loss()

onert::ir::operation::Loss::Loss ( const OperandIndexSequence &  inputs,
const OperandIndexSequence &  outputs 
)

Definition at line 31 of file Loss.cc.

32 : Operation{OperandConstraint::createAtLeast(2u), inputs, outputs}
33{
34 assert(inputs.size() == 2);
35}
onert::ir::OperandConstraint::createAtLeast
static OperandConstraint createAtLeast(uint32_t begin)
Definition OperandConstraint.h:38

Member Function Documentation

◆ accept()

void onert::ir::operation::Loss::accept ( OperationVisitor &  v) const
overridevirtual

Implements onert::ir::IOperation.

Reimplemented in onert::ir::train::operation::Loss.

Definition at line 29 of file Loss.cc.

29{ v.visit(*this); }

◆ opcode()

OpCode onert::ir::operation::Loss::opcode ( ) const
inlinefinalvirtual

Implements onert::ir::IOperation.

Definition at line 44 of file Loss.h.

44{ return OpCode::Loss; }

Referenced by onert::ir::train::operation::Loss::name().

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