test_topi_transform.py
22.5 KB
-
[Relay/TOPI][Op] Add shape op in Relay and TOPI (#2749) · ee805806* Add shapeof op in topi * Add relay shape_of op * Add constant folding for shape_of * Allow shape op to specify dtype * Add mxnet converter for shape_array * lint * lint * Add doc
Haichen Shen committed