#undef TENSORFLOW_NODE
protected:
+ const locop::SymbolTable *tbl(void) const { return _tbl; }
+
+ // Please do not use _tbl directly and use tbl().
+ // This will be changed to private in near future.
+protected:
const locop::SymbolTable *_tbl;
};
bool TFNodeSummaryBuilder::summary(const TFAdd *node, locop::NodeSummary &s) const
{
s.opname("TF.Add");
- s.args().append("x", _tbl->lookup(node->x()));
- s.args().append("y", _tbl->lookup(node->y()));
+ s.args().append("x", tbl()->lookup(node->x()));
+ s.args().append("y", tbl()->lookup(node->y()));
s.state(locop::NodeSummary::State::Complete);
return true;
}
bool TFNodeSummaryBuilder::summary(const TFAvgPool *node, locop::NodeSummary &s) const
{
s.opname("TF.AvgPool");
- s.args().append("value", _tbl->lookup(node->value()));
+ s.args().append("value", tbl()->lookup(node->value()));
s.args().append("ksize", pepper::str(node->ksize()));
s.args().append("strides", pepper::str(node->strides()));
s.args().append("padding", node->padding());
bool TFNodeSummaryBuilder::summary(const TFBiasAdd *node, locop::NodeSummary &s) const
{
s.opname("TF.BiasAdd");
- s.args().append("value", _tbl->lookup(node->value()));
- s.args().append("bias", _tbl->lookup(node->bias()));
+ s.args().append("value", tbl()->lookup(node->value()));
+ s.args().append("bias", tbl()->lookup(node->bias()));
s.args().append("data_layout", node->data_layout());
s.state(locop::NodeSummary::State::Complete);
return true;
bool TFNodeSummaryBuilder::summary(const TFConv2D *node, locop::NodeSummary &s) const
{
s.opname("TF.Conv2D");
- s.args().append("ifm", _tbl->lookup(node->ifm()));
- s.args().append("ker", _tbl->lookup(node->ker()));
+ s.args().append("ifm", tbl()->lookup(node->ifm()));
+ s.args().append("ker", tbl()->lookup(node->ker()));
s.args().append("padding", node->padding());
s.args().append("data_layout", node->data_layout());
s.args().append("strides", pepper::str(node->strides()));
bool TFNodeSummaryBuilder::summary(const TFDepthwiseConv2dNative *node, locop::NodeSummary &s) const
{
s.opname("TF.DepthwiseConv2dNative");
- s.args().append("ifm", _tbl->lookup(node->ifm()));
- s.args().append("ker", _tbl->lookup(node->ker()));
+ s.args().append("ifm", tbl()->lookup(node->ifm()));
+ s.args().append("ker", tbl()->lookup(node->ker()));
s.args().append("padding", node->padding());
s.args().append("data_layout", node->data_layout());
s.args().append("strides", pepper::str(node->strides()));
bool TFNodeSummaryBuilder::summary(const TFFusedBatchNorm *node, locop::NodeSummary &s) const
{
s.opname("TF.FusedBatchNorm");
- s.args().append("input", _tbl->lookup(node->input()));
- s.args().append("gamma", _tbl->lookup(node->gamma()));
- s.args().append("beta", _tbl->lookup(node->beta()));
- s.args().append("mean", _tbl->lookup(node->mean()));
- s.args().append("variance", _tbl->lookup(node->variance()));
+ s.args().append("input", tbl()->lookup(node->input()));
+ s.args().append("gamma", tbl()->lookup(node->gamma()));
+ s.args().append("beta", tbl()->lookup(node->beta()));
+ s.args().append("mean", tbl()->lookup(node->mean()));
+ s.args().append("variance", tbl()->lookup(node->variance()));
s.args().append("epsilon", pepper::str(node->epsilon()));
s.state(locop::NodeSummary::State::Complete);
return true;
bool TFNodeSummaryBuilder::summary(const TFMaxPool *node, locop::NodeSummary &s) const
{
s.opname("TF.MaxPool");
- s.args().append("value", _tbl->lookup(node->value()));
+ s.args().append("value", tbl()->lookup(node->value()));
s.args().append("ksize", pepper::str(node->ksize()));
s.args().append("strides", pepper::str(node->strides()));
s.args().append("padding", node->padding());
bool TFNodeSummaryBuilder::summary(const TFMul *node, locop::NodeSummary &s) const
{
s.opname("TF.Mul");
- s.args().append("x", _tbl->lookup(node->x()));
- s.args().append("y", _tbl->lookup(node->y()));
+ s.args().append("x", tbl()->lookup(node->x()));
+ s.args().append("y", tbl()->lookup(node->y()));
s.state(locop::NodeSummary::State::Complete);
return true;
}
bool TFNodeSummaryBuilder::summary(const TFReshape *node, locop::NodeSummary &s) const
{
s.opname("TF.Reshape");
- s.args().append("tensor", _tbl->lookup(node->tensor()));
- s.args().append("shape", _tbl->lookup(node->shape()));
+ s.args().append("tensor", tbl()->lookup(node->tensor()));
+ s.args().append("shape", tbl()->lookup(node->shape()));
s.state(locop::NodeSummary::State::Complete);
return true;
}
bool TFNodeSummaryBuilder::summary(const TFRsqrt *node, locop::NodeSummary &s) const
{
s.opname("TF.Rsqrt");
- s.args().append("x", _tbl->lookup(node->x()));
+ s.args().append("x", tbl()->lookup(node->x()));
s.state(locop::NodeSummary::State::Complete);
return true;
}