| /external/ComputeLibrary/tests/validation/dynamic_fusion/gpu/ |
| D | Integration.cpp | 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()
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()
[all …]
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/ |
| D | GpuReshape.cpp | 88 Status GpuReshape::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
106 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in validate_op()
107 ARM_COMPUTE_RETURN_ERROR_ON_MSG(!sketch.implementation().operator_group().try_add_operator(op), in validate_op()
111 return is_supported_op_helper(*sketch.gpu_context(), src, &dst_info_to_validate, attributes); in validate_op()
114 ITensorInfo *GpuReshape::create_op(GpuWorkloadSketch &sketch, in create_op() argument
120 ARM_COMPUTE_ERROR_THROW_ON(GpuReshape::validate_op(sketch, src, attributes)); in create_op()
122 ITensorInfo *dst = sketch.implementation().create_virtual_tensor(); in create_op()
128 auto &comp_graph = sketch.implementation().component_graph(); in create_op()
129 const auto sketch_ctx = sketch.implementation().context(); in create_op()
158 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in create_op()
[all …]
|
| D | GpuClamp.cpp | 97 Status GpuClamp::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
116 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in validate_op()
117 ARM_COMPUTE_RETURN_ERROR_ON_MSG(!sketch.implementation().operator_group().try_add_operator(op), in validate_op()
121 return is_supported_op_helper(*sketch.gpu_context(), src, &dst_info_to_validate, attributes); in validate_op()
124 ITensorInfo *GpuClamp::create_op(GpuWorkloadSketch &sketch, in create_op() argument
130 ARM_COMPUTE_ERROR_THROW_ON(GpuClamp::validate_op(sketch, src, attributes)); in create_op()
132 ITensorInfo *dst = sketch.implementation().create_virtual_tensor(); in create_op()
139 GpuKernelComponentGraph &comp_graph = sketch.implementation().component_graph(); in create_op()
147 const auto *const sketch_ctx = sketch.implementation().context(); in create_op()
173 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in create_op()
[all …]
|
| D | GpuSoftmax.cpp | 100 Status GpuSoftmax::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
123 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in validate_op()
124 ARM_COMPUTE_RETURN_ERROR_ON_MSG(!sketch.implementation().operator_group().try_add_operator(op), in validate_op()
128 return is_supported_op(*sketch.gpu_context(), src, &dst_info_to_validate, attributes); in validate_op()
131 void GpuSoftmax::create_op(GpuWorkloadSketch &sketch, in create_op() argument
139 …ITensorInfo *logits = sketch.implementation().create_auxiliary_tensor(src->clone()->set_… in create_op()
141 …ITensorInfo *sum = sketch.implementation().create_auxiliary_tensor(src->clone()->set_tensor_shape(… in create_op()
147 ARM_COMPUTE_ERROR_THROW_ON(GpuSoftmax::validate_op(sketch, src, dst, attributes)); in create_op()
151 auto &comp_graph = sketch.implementation().component_graph(); in create_op()
152 const auto sketch_ctx = sketch.implementation().context(); in create_op()
[all …]
|
| D | GpuCast.cpp | 104 Status GpuCast::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
122 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in validate_op()
123 ARM_COMPUTE_RETURN_ERROR_ON_MSG(!sketch.implementation().operator_group().try_add_operator(op), in validate_op()
127 return is_supported_op_helper(*sketch.gpu_context(), src, &dst_info_to_validate, attributes); in validate_op()
130 ITensorInfo *GpuCast::create_op(GpuWorkloadSketch &sketch, in create_op() argument
136 ARM_COMPUTE_ERROR_THROW_ON(GpuCast::validate_op(sketch, src, attributes)); in create_op()
138 ITensorInfo *dst = sketch.implementation().create_virtual_tensor(); in create_op()
145 GpuKernelComponentGraph &comp_graph = sketch.implementation().component_graph(); in create_op()
146 const auto *sketch_ctx = sketch.implementation().context(); in create_op()
176 … const Operator op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in create_op()
[all …]
|
| D | GpuResize.cpp | 117 Status GpuResize::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
135 … const Operator op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in validate_op()
137 ARM_COMPUTE_RETURN_ERROR_ON_MSG(!sketch.implementation().operator_group().try_add_operator(op), in validate_op()
141 return is_supported_op_helper(*sketch.gpu_context(), src, &dst_info_to_validate, attributes); in validate_op()
144 ITensorInfo *GpuResize::create_op(GpuWorkloadSketch &sketch, in create_op() argument
150 ARM_COMPUTE_ERROR_THROW_ON(GpuResize::validate_op(sketch, src, attributes)); in create_op()
152 ITensorInfo *dst = sketch.implementation().create_virtual_tensor(); in create_op()
159 GpuKernelComponentGraph &comp_graph = sketch.implementation().component_graph(); in create_op()
160 const auto *sketch_ctx = sketch.implementation().context(); in create_op()
189 … const Operator op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in create_op()
[all …]
|
| D | GpuOutput.cpp | 63 Status GpuOutput::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
83 const auto group = sketch.implementation().operator_group(); in validate_op()
89 const auto status = is_supported_op(*sketch.gpu_context(), src, dst); in validate_op()
93 void GpuOutput::create_op(GpuWorkloadSketch &sketch, in create_op() argument
98 ARM_COMPUTE_ERROR_THROW_ON(GpuOutput::validate_op(sketch, src, dst)); in create_op()
104 auto &comp_graph = sketch.implementation().component_graph(); in create_op()
105 const auto sketch_ctx = sketch.implementation().context(); in create_op()
135 … const Operator op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in create_op()
136 sketch.implementation().operator_group().add_operator(op, true); in create_op()
|
| D | GpuPool2d.cpp | 63 Status GpuPool2d::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
85 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in validate_op()
86 ARM_COMPUTE_RETURN_ERROR_ON_MSG(!sketch.implementation().operator_group().try_add_operator(op), in validate_op()
90 return is_supported_op(*sketch.gpu_context(), src, &dst_info_to_validate, attributes, settings); in validate_op()
137 void GpuPool2d::create_op(GpuWorkloadSketch &sketch, in create_op() argument
144 ARM_COMPUTE_ERROR_THROW_ON(GpuPool2d::validate_op(sketch, src, dst, attributes, settings)); in create_op()
154 auto &comp_graph = sketch.implementation().component_graph(); in create_op()
156 const auto sketch_ctx = sketch.implementation().context(); in create_op()
187 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in create_op()
188 sketch.implementation().operator_group().add_operator(op); in create_op()
|
| D | GpuDepthwiseConv2d.cpp | 150 Status GpuDepthwiseConv2d::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
177 … const Operator op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in validate_op()
179 ARM_COMPUTE_RETURN_ERROR_ON_MSG(!sketch.implementation().operator_group().try_add_operator(op), in validate_op()
183 …return is_supported_op_helper(*sketch.gpu_context(), src, wei, bia, &dst_info_to_validate, attribu… in validate_op()
186 ITensorInfo *GpuDepthwiseConv2d::create_op(GpuWorkloadSketch &sketch, in create_op() argument
194 ARM_COMPUTE_ERROR_THROW_ON(GpuDepthwiseConv2d::validate_op(sketch, src, wei, bia, attributes)); in create_op()
196 ITensorInfo *dst = sketch.implementation().create_virtual_tensor(); in create_op()
202 GpuKernelComponentGraph &comp_graph = sketch.implementation().component_graph(); in create_op()
203 const auto *sketch_ctx = sketch.implementation().context(); in create_op()
262 … const Operator op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in create_op()
[all …]
|
| D | GpuConv2d.cpp | 188 Status GpuConv2d::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
219 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in validate_op()
220 ARM_COMPUTE_RETURN_ERROR_ON_MSG(!sketch.implementation().operator_group().try_add_operator(op), in validate_op()
224 …return is_supported_op_helper(*sketch.gpu_context(), src, wei, bia, &dst_info_to_validate, attribu… in validate_op()
227 ITensorInfo *GpuConv2d::create_op(GpuWorkloadSketch &sketch, in create_op() argument
240 ITensorInfo *dst = sketch.implementation().create_virtual_tensor(); in create_op()
243 ARM_COMPUTE_ERROR_THROW_ON(GpuConv2d::validate_op(sketch, src, wei, bia, attributes)); in create_op()
250 auto &comp_graph = sketch.implementation().component_graph(); in create_op()
252 const auto sketch_ctx = sketch.implementation().context(); in create_op()
306 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in create_op()
[all …]
|
| D | GpuMul.cpp | 35 Status GpuMul::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
46 return GpuElementwiseBinaryCommon::validate_op(sketch, lhs, rhs, common_attributes); in validate_op()
63 ITensorInfo *GpuMul::create_op(GpuWorkloadSketch &sketch, in create_op() argument
70 return GpuElementwiseBinaryCommon::create_op(sketch, lhs, rhs, common_attributes); in create_op()
|
| D | GpuAdd.cpp | 35 Status GpuAdd::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
46 return GpuElementwiseBinaryCommon::validate_op(sketch, lhs, rhs, common_attributes); in validate_op()
63 ITensorInfo *GpuAdd::create_op(GpuWorkloadSketch &sketch, in create_op() argument
71 return GpuElementwiseBinaryCommon::create_op(sketch, lhs, rhs, common_attributes); in create_op()
|
| D | GpuSub.cpp | 35 Status GpuSub::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
46 return GpuElementwiseBinaryCommon::validate_op(sketch, lhs, rhs, common_attributes); in validate_op()
63 ITensorInfo *GpuSub::create_op(GpuWorkloadSketch &sketch, in create_op() argument
71 return GpuElementwiseBinaryCommon::create_op(sketch, lhs, rhs, common_attributes); in create_op()
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/internal/ |
| D | GpuElementwiseBinaryCommon.cpp | 112 Status GpuElementwiseBinaryCommon::validate_op(const GpuWorkloadSketch &sketch, in validate_op() argument
132 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in validate_op()
133 ARM_COMPUTE_RETURN_ERROR_ON_MSG(!sketch.implementation().operator_group().try_add_operator(op), in validate_op()
137 … return is_supported_op_helper(*sketch.gpu_context(), lhs, rhs, &dst_info_to_validate, attributes); in validate_op()
140 ITensorInfo *GpuElementwiseBinaryCommon::create_op(GpuWorkloadSketch &sketch, in create_op() argument
147 … ARM_COMPUTE_ERROR_THROW_ON(GpuElementwiseBinaryCommon::validate_op(sketch, lhs, rhs, attributes)); in create_op()
149 ITensorInfo *dst = sketch.implementation().create_virtual_tensor(); in create_op()
156 auto &comp_graph = sketch.implementation().component_graph(); in create_op()
158 const auto sketch_ctx = sketch.implementation().context(); in create_op()
189 const auto op = sketch.implementation().operator_group().new_operator(operator_type, tensors); in create_op()
[all …]
|
| /external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/operators/ |
| D | MulFixture.h | 79 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
82 TensorInfo lhs_info = sketch.create_tensor_info(TensorInfo(shape0, 1, _data_type)); in compute_target()
83 TensorInfo rhs_info = sketch.create_tensor_info(TensorInfo(shape1, 1, _data_type)); in compute_target()
84 TensorInfo dst_info = sketch.create_tensor_info(); in compute_target()
88 ITensorInfo *ans_info = FunctionType::create_op(sketch, &lhs_info, &rhs_info); in compute_target()
92 rhs_info_fuse = sketch.create_tensor_info(TensorInfo(shape2, 1, _data_type)); in compute_target()
93 ITensorInfo *ans2_info = FunctionType::create_op(sketch, ans_info, &rhs_info_fuse); in compute_target()
94 GpuOutput::create_op(sketch, ans2_info, &dst_info); in compute_target()
98 GpuOutput::create_op(sketch, ans_info, &dst_info); in compute_target()
103 runtime.configure(sketch); in compute_target()
|
| D | ClampFixture.h | 108 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
111 TensorInfo src_info = sketch.create_tensor_info(TensorInfo(shape, 1, _data_type)); in compute_target()
112 TensorInfo dst_info = sketch.create_tensor_info(TensorInfo(shape, 1, _data_type)); in compute_target()
114 ITensorInfo *ans_0_info = FunctionType::create_op(sketch, &src_info, attributes); in compute_target()
117 ITensorInfo *ans_1_info = FunctionType::create_op(sketch, ans_0_info, attributes); in compute_target()
118 GpuOutput::create_op(sketch, ans_1_info, &dst_info); in compute_target()
122 GpuOutput::create_op(sketch, ans_0_info, &dst_info); in compute_target()
127 runtime.configure(sketch); in compute_target()
|
| D | ReshapeFixture.h | 75 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
78 … TensorInfo src_info = sketch.create_tensor_info(TensorInfo(input_shape, 1, data_type)); in compute_target()
79 … TensorInfo dst_info = sketch.create_tensor_info(TensorInfo(output_shape, 1, data_type)); in compute_target()
83 ITensorInfo *ans_info = FunctionType::create_op(sketch, &src_info, attributes); in compute_target()
84 GpuOutput::create_op(sketch, ans_info, &dst_info); in compute_target()
88 runtime.configure(sketch); in compute_target()
|
| D | CastFixture.h | 116 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
119 …TensorInfo src_info = sketch.create_tensor_info(TensorInfo(shape, 1, dt_in, DataLayout::NCHW)); //… in compute_target()
120 TensorInfo dst_info = sketch.create_tensor_info(); in compute_target()
125 ITensorInfo *ans_info = FunctionType::create_op(sketch, &src_info, attributes); in compute_target()
126 GpuOutput::create_op(sketch, ans_info, &dst_info); in compute_target()
130 runtime.configure(sketch); in compute_target()
|
| D | SoftmaxFixture.h | 86 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
90 TensorInfo src_info = sketch.create_tensor_info(shape, 1, data_type); in compute_target()
91 TensorInfo dst_info = sketch.create_tensor_info(shape, 1, data_type); in compute_target()
92 FunctionType::create_op(sketch, &src_info, &dst_info, softmax_attr); in compute_target()
96 runtime.configure(sketch); in compute_target()
|
| D | ResizeFixture.h | 141 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
144 … TensorInfo src_info = sketch.create_tensor_info(TensorInfo(shape, 1, _data_type, _data_layout)); in compute_target()
146 TensorInfo dst_info = sketch.create_tensor_info(); in compute_target()
151 ITensorInfo *scale_result_info = FunctionType::create_op(sketch, &src_info, attributes); in compute_target()
152 GpuOutput::create_op(sketch, scale_result_info, &dst_info); in compute_target()
156 runtime.configure(sketch); in compute_target()
|
| /external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/ |
| D | DirectConv2dFixture.h | 119 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
122 …TensorInfo input_info = sketch.create_tensor_info(TensorInfo(input_shape, 1, _data_type, _data_la… in compute_target()
123 …TensorInfo weight_info = sketch.create_tensor_info(TensorInfo(weights_shape, 1, _data_type, _data_… in compute_target()
124 …TensorInfo bias_info = sketch.create_tensor_info(TensorInfo(bias_shape, 1, _data_type, _data_lay… in compute_target()
125 TensorInfo dst_info = sketch.create_tensor_info(); in compute_target()
127 …ITensorInfo *ans_info = FunctionType::create_op(sketch, &input_info, &weight_info, &bias_info, con… in compute_target()
128 GpuOutput::create_op(sketch, ans_info, &dst_info); in compute_target()
132 runtime.configure(sketch); in compute_target()
260 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
263 … auto input_info = sketch.create_tensor_info(TensorInfo(input_shape, 1, data_type, data_layout)); in compute_target()
[all …]
|
| D | ElementwiseBinaryFixture.h | 103 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
106 TensorInfo lhs_info = sketch.create_tensor_info(TensorInfo(shape0, 1, _data_type)); in compute_target()
107 TensorInfo rhs_info = sketch.create_tensor_info(TensorInfo(shape1, 1, _data_type)); in compute_target()
108 TensorInfo dst_info = sketch.create_tensor_info(); in compute_target()
112 ITensorInfo *ans_info = FunctionType::create_op(sketch, &lhs_info, &rhs_info); in compute_target()
116 rhs_info_fuse = sketch.create_tensor_info(TensorInfo(shape2, 1, _data_type)); in compute_target()
117 ITensorInfo *ans2_info = FunctionType::create_op(sketch, ans_info, &rhs_info_fuse); in compute_target()
118 GpuOutput::create_op(sketch, ans2_info, &dst_info); in compute_target()
122 GpuOutput::create_op(sketch, ans_info, &dst_info); in compute_target()
127 runtime.configure(sketch); in compute_target()
|
| D | DepthwiseConv2dFixture.h | 130 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
133 …TensorInfo input_info = sketch.create_tensor_info(TensorInfo(input_shape, 1, _data_type, _data_la… in compute_target()
134 …TensorInfo weight_info = sketch.create_tensor_info(TensorInfo(weights_shape, 1, _data_type, _data_… in compute_target()
135 …TensorInfo bias_info = sketch.create_tensor_info(TensorInfo(bias_shape, 1, _data_type, _data_lay… in compute_target()
136 TensorInfo dst_info = sketch.create_tensor_info(); in compute_target()
138 …ITensorInfo *ans_info = FunctionType::create_op(sketch, &input_info, &weight_info, &bias_info, dwc… in compute_target()
139 GpuOutput::create_op(sketch, ans_info, &dst_info); in compute_target()
143 runtime.configure(sketch); in compute_target()
|
| D | Pool2dFixture.h | 95 GpuWorkloadSketch sketch{ &gpu_ctx }; in compute_target()
98 …auto input_info = sketch.create_tensor_info(TensorInfo(input_shape, 1, data_type, DataLayout::NHWC… in compute_target()
99 auto dst_info = sketch.create_tensor_info(); in compute_target()
104 FunctionType::create_op(sketch, &input_info, &dst_info, pool_attr, pool_settings); in compute_target()
108 runtime.configure(sketch); in compute_target()
|
| /external/ComputeLibrary/tests/validation/dynamic_fusion/gpu/cl/ |
| D | Resize.cpp | 99 GpuWorkloadSketch sketch{ &gpu_ctx }; in TEST_CASE() local
101 const TensorInfo sketch_input_info = sketch.create_tensor_info(input_info); in TEST_CASE()
104 Status status = GpuResize::validate_op(sketch, nullptr, ResizeAttributes()); in TEST_CASE()
139 GpuWorkloadSketch sketch{ &gpu_ctx }; in TEST_CASE() local
141 const TensorInfo sketch_input_info = sketch.create_tensor_info(input_info); in TEST_CASE()
147 Status status = GpuResize::validate_op(sketch, &sketch_input_info, attributes); in TEST_CASE()
161 GpuWorkloadSketch sketch{ &gpu_ctx }; in TEST_CASE() local
163 const TensorInfo sketch_input_info = sketch.create_tensor_info(input_info); in TEST_CASE()
165 Status status = GpuResize::validate_op(sketch, &sketch_input_info, ResizeAttributes()); in TEST_CASE()
181 GpuWorkloadSketch sketch{ &gpu_ctx }; in TEST_CASE() local
[all …]
|