ONE - On-device Neural Engine
|
GraphBuilder for Squeeze node. More...
#include <Squeeze.h>
Public Member Functions | |
bool | validate (const tensorflow::NodeDef &) const override |
void | build (const tensorflow::NodeDef &, GraphBuilderContext *) const override |
Public Member Functions inherited from moco::GraphBuilder | |
virtual | ~GraphBuilder () |
|
overridevirtual |
Implements moco::GraphBuilder.
Definition at line 80 of file Squeeze.cpp.
References plier::tf::as_int64_list(), moco::SymbolTable::enroll(), moco::UpdateQueue::enroll(), plier::tf::get_list_attr(), moco::GraphBuilderContext::graph(), plier::tf::has_attrs(), moco::TFNode::name(), moco::GraphBuilderContext::tensor_names(), moco::update(), and moco::GraphBuilderContext::updates().
|
overridevirtual |
Implements moco::GraphBuilder.
Definition at line 63 of file Squeeze.cpp.
References plier::tf::has_attrs().