Lines Matching full:sketch
28 #include "arm_compute/dynamic_fusion/sketch/attributes/CastAttributes.h"
29 #include "arm_compute/dynamic_fusion/sketch/attributes/Conv2dAttributes.h"
30 #include "arm_compute/dynamic_fusion/sketch/gpu/GpuWorkloadSketch.h"
31 #include "arm_compute/dynamic_fusion/sketch/gpu/operators/GpuAdd.h"
32 #include "arm_compute/dynamic_fusion/sketch/gpu/operators/GpuCast.h"
33 #include "arm_compute/dynamic_fusion/sketch/gpu/operators/GpuConv2d.h"
34 #include "arm_compute/dynamic_fusion/sketch/gpu/operators/GpuOutput.h"
70 // Create a new workload sketch in TEST_SUITE()
73 GpuWorkloadSketch sketch{ &gpu_ctx }; in TEST_SUITE() local
77 …TensorInfo input_info = sketch.create_tensor_info(t_input_shape, 1, data_type, data_layout); in TEST_SUITE()
78 …TensorInfo weight_info = sketch.create_tensor_info(TensorInfo(t_weight_shape, 1, data_type, … in TEST_SUITE()
80 …ITensorInfo *conv_out_info = GpuConv2d::create_op(sketch, &input_info, &weight_info, nullptr, conv… in TEST_SUITE()
82 TensorInfo dst_info = sketch.create_tensor_info(); in TEST_SUITE()
83 GpuOutput::create_op(sketch, conv_out_info, &dst_info); in TEST_SUITE()
87 runtime.configure(sketch); in TEST_SUITE()
157 // Create a new workload sketch in TEST_CASE()
160 GpuWorkloadSketch sketch{ &gpu_ctx }; in TEST_CASE() local
162 TensorInfo in_0_info = sketch.create_tensor_info(t_input_shape, 1, data_type); in TEST_CASE()
163 TensorInfo in_1_info = sketch.create_tensor_info(t_input_shape, 1, data_type); in TEST_CASE()
164 TensorInfo in_2_info = sketch.create_tensor_info(t_input_shape, 1, data_type); in TEST_CASE()
166 TensorInfo out_0_info = sketch.create_tensor_info(); in TEST_CASE()
167 TensorInfo out_1_info = sketch.create_tensor_info(); in TEST_CASE()
169 ITensorInfo *ans_0_info = GpuAdd::create_op(sketch, &in_0_info, &in_1_info); in TEST_CASE()
170 GpuOutput::create_op(sketch, ans_0_info, &out_0_info); in TEST_CASE()
171 ITensorInfo *ans_1_info = GpuAdd::create_op(sketch, ans_0_info, &in_2_info); in TEST_CASE()
172 GpuOutput::create_op(sketch, ans_1_info, &out_1_info); in TEST_CASE()
176 runtime.configure(sketch); in TEST_CASE()
254 // Create a new workload sketch in TEST_CASE()
257 GpuWorkloadSketch sketch{ &gpu_ctx }; in TEST_CASE() local
259 TensorInfo in_0_info = sketch.create_tensor_info(t_input_shape, 1, data_type); in TEST_CASE()
260 TensorInfo in_1_info = sketch.create_tensor_info(t_input_shape, 1, data_type); in TEST_CASE()
261 TensorInfo in_2_info = sketch.create_tensor_info(t_input_shape, 1, data_type); in TEST_CASE()
263 TensorInfo out_0_info = sketch.create_tensor_info(); in TEST_CASE()
264 TensorInfo out_1_info = sketch.create_tensor_info(); in TEST_CASE()
272 ITensorInfo *ans_0_info = GpuAdd::create_op(sketch, &in_0_info, &in_1_info); in TEST_CASE()
273 GpuOutput::create_op(sketch, ans_0_info, &out_0_info); in TEST_CASE()
274 ITensorInfo *ans_1_info = GpuAdd::create_op(sketch, ans_0_info, &in_2_info); in TEST_CASE()
275 ITensorInfo *ans_2_info = GpuCast::create_op(sketch, ans_1_info, cast_0_attr); in TEST_CASE()
276 ITensorInfo *ans_3_info = GpuCast::create_op(sketch, ans_2_info, cast_1_attr); in TEST_CASE()
277 GpuOutput::create_op(sketch, ans_3_info, &out_1_info); in TEST_CASE()
281 runtime.configure(sketch); in TEST_CASE()
369 // Create a new workload sketch in TEST_CASE()
372 GpuWorkloadSketch sketch{ &gpu_ctx }; in TEST_CASE() local
375 TensorInfo input_info = sketch.create_tensor_info(t_input_shape, 1, data_type, data_layout); in TEST_CASE()
376 …TensorInfo weight_info = sketch.create_tensor_info(TensorInfo(t_weight_shape, 1, data_type, data… in TEST_CASE()
382 …const Status success = GpuConv2d::validate_op(sketch, &input_info, &weight_info, nullptr, conv2d_a… in TEST_CASE()
385 dst_info = GpuConv2d::create_op(sketch, &input_info, &weight_info, nullptr, conv2d_attr); in TEST_CASE()
389 TensorInfo weight_info_2 = sketch.create_tensor_info(t_weight_info); in TEST_CASE()
394 …const Status success = GpuConv2d::validate_op(sketch, dst_info, &weight_info_2, nullptr… in TEST_CASE()