ONE - On-device Neural Engine
|
GraphBuilder for StopGradient node. More...
#include <StopGradient.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 () |
GraphBuilder for StopGradient node.
Definition at line 28 of file StopGradient.h.
|
overridevirtual |
Implements moco::GraphBuilder.
Definition at line 65 of file StopGradient.cpp.
References moco::SymbolTable::enroll(), moco::UpdateQueue::enroll(), moco::GraphBuilderContext::graph(), moco::TFNode::name(), moco::GraphBuilderContext::tensor_names(), and moco::GraphBuilderContext::updates().
|
overridevirtual |
Implements moco::GraphBuilder.
Definition at line 57 of file StopGradient.cpp.
References plier::tf::has_attrs().