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

#include <CircleGatherNd.h>

Collaboration diagram for luci::CircleGatherNdGraphBuilder:

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 CircleGatherNd.h.

Member Function Documentation

◆ validate()

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

Implements luci::GraphBuilderBase.

Definition at line 28 of file CircleGatherNd.cpp.

29{
30 if (!GraphBuilder::validate(args, 2))
31 return false;
32
33 const auto &inputs = args.op.inputs;
34 auto indices_tensor = args.reader.tensors()[inputs.at(1)];
35 assert(indices_tensor != nullptr);
36
37 if (!(indices_tensor->type() == circle::TensorType::TensorType_INT32 ||
38 indices_tensor->type() == circle::TensorType::TensorType_INT64))
39 {
40 return false;
41 }
42
43 return true;
44}
bool validate(const ValidateArgs &args, size_t input_cnt) const
args
Definition infer.py:21

References luci::GraphBuilder::validate().


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