MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / SmallReshapeLayerDataset

Class SmallReshapeLayerDataset

tests/datasets/ReshapeLayerDataset.h:102–116  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

100/** [ReshapeLayer datasets] **/
101
102class SmallReshapeLayerDataset final : public ReshapeLayerDataset
103{
104public:
105 SmallReshapeLayerDataset()
106 {
107 add_config(TensorShape(3U), TensorShape(1U, 1U, 3U));
108 add_config(TensorShape(16U), TensorShape(4U, 2U, 2U));
109 add_config(TensorShape(2U, 2U, 8U), TensorShape(4U, 8U));
110 add_config(TensorShape(3U, 3U, 16U), TensorShape(144U));
111 add_config(TensorShape(17U, 3U, 12U), TensorShape(1U, 1U, 612U));
112 add_config(TensorShape(26U, 26U, 32U), TensorShape(13U, 13U, 128U));
113 add_config(TensorShape(31U, 23U, 4U, 7U), TensorShape(2U, 14U, 713U));
114 add_config(TensorShape(8U, 8U, 8U), TensorShape(8U, 64U));
115 }
116};
117} // namespace datasets
118} // namespace test
119} // namespace arm_compute

Callers 2

ReshapeLayer.cppFile · 0.85
ReshapeLayer.cppFile · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected