colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
TIM-VX/src/tim/transform/ops
History
liyuenan 748274143b
support layout inference for operations (#39) 
Add layout inference support for space2depth, depth2space, space2batch, batch2space, pad and
reduce.

Signed-off-by: yuenan.li <yuenan.li@verisilicon.com>

Co-authored-by: yuenan.li <yuenan.li@verisilicon.com>
 		2021-05-13 22:27:23 +08:00
..
activation_layout_inference.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00
batch2space_layout_inference.h support layout inference for operations (#39) 2021-05-13 22:27:23 +08:00
concat_layout_inferene.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00
conv2d_layout_inference.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00
depth2space_layout_inference.h support layout inference for operations (#39) 2021-05-13 22:27:23 +08:00
elementwise_layout_inference.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00
op_layout_inference.cc support layout inference for operations (#39) 2021-05-13 22:27:23 +08:00
op_layout_inference.h support layout inference for operations (#39) 2021-05-13 22:27:23 +08:00
pad_layout_inference.h support layout inference for operations (#39) 2021-05-13 22:27:23 +08:00
pool2d_layout_inference.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00
reduce_layout_inference.h support layout inference for operations (#39) 2021-05-13 22:27:23 +08:00
reshape_layout_inference.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00
simple_ops_layout_inference.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00
softmax_layout_inference.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00
space2batch_layout_inference.h support layout inference for operations (#39) 2021-05-13 22:27:23 +08:00
space2depth_layout_inference.h support layout inference for operations (#39) 2021-05-13 22:27:23 +08:00
squeeze_layout_inference.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00
stack_layout_inference.h Align directory name to namespace for layout inference (#38) 2021-05-11 09:46:46 +08:00