ONE - On-device Neural Engine
Loading...
Searching...
No Matches
luci::CircleEluGraphBuilder Class Reference

#include <CircleElu.h>

Collaboration diagram for luci::CircleEluGraphBuilder:

Public Member Functions

bool validate (const ValidateArgs &args) const final
 
- Public Member Functions inherited from luci::GraphBuilder
virtual ~GraphBuilder ()=default
 
bool validate (const ValidateArgs &args, size_t input_cnt) const
 
CircleNodebuild (const circle::OperatorT &op, GraphBuilderContext *context) const final
 
- Public Member Functions inherited from luci::GraphBuilderBase
virtual ~GraphBuilderBase ()=default
 

Detailed Description

Definition at line 25 of file CircleElu.h.

Member Function Documentation

◆ validate()

bool luci::CircleEluGraphBuilder::validate ( const ValidateArgs args) const
finalvirtual

Implements luci::GraphBuilderBase.

Definition at line 26 of file CircleElu.cpp.

27{
28 if (!GraphBuilder::validate(args, 1))
29 return false;
30
31 const auto &inputs = args.op.inputs;
32 const auto &outputs = args.op.outputs;
33
34 const auto tensors = args.reader.tensors();
35 const auto tensor = tensors.at(inputs.at(0));
36 assert(tensor != nullptr);
37
38 switch (tensor->type())
39 {
40 case circle::TensorType_FLOAT64:
41 break;
42 case circle::TensorType_FLOAT32:
43 break;
44 case circle::TensorType_INT16:
45 break;
46 case circle::TensorType_UINT8:
47 break;
48 default:
49 return false;
50 }
51
52 assert(tensors[outputs[0]] != nullptr);
53 if (tensors[outputs[0]]->type() != tensor->type())
54 return false;
55
56 return true;
57}
bool validate(const ValidateArgs &args, size_t input_cnt) const
args
Definition infer.py:21
type
Definition infer.py:18

References luci::GraphBuilder::validate().


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