CanonicalNodeSummaryBuilder now ignores (deprecated) tensor shape
attributes when it generates a node summary for Push nodes.
Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
return res;
}
-loco::TensorShape push_shape(const loco::Push *m)
-{
- loco::TensorShape res;
-
- res.rank(m->rank());
-
- for (uint32_t axis = 0; axis < m->rank(); ++axis)
- {
- res.dim(axis) = m->dim(axis);
- }
-
- return res;
-}
-
loco::TensorShape pull_shape(const loco::Pull *m)
{
loco::TensorShape res;
res.arg("index", node->indexed() ? pp::fmt(node->index()) : pp::fmt('?'));
res.arg("from", symbol_lookup(*_symtbl, node->from()));
- res.arg("shape", pp::fmt(pretty(push_shape(node))));
res.state(NodeDesc::State::Complete);
return res;