Home
last modified time | relevance | path

Searched refs:get_axis_size (Results 1 – 24 of 24) sorted by relevance

/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/csl/
H A Dtensor_ops.hpp52 CV_Assert(A.get_axis_size(-2) == result.get_axis_size(-2)); in assertGEMMCompatiblity()
53 CV_Assert(A.get_axis_size(-1) == B.get_axis_size(-2)); in assertGEMMCompatiblity()
54 CV_Assert(B.get_axis_size(-1) == result.get_axis_size(-1)); in assertGEMMCompatiblity()
56 CV_Assert(A.get_axis_size(-2) == result.get_axis_size(-2)); in assertGEMMCompatiblity()
57 CV_Assert(A.get_axis_size(-1) == B.get_axis_size(-1)); in assertGEMMCompatiblity()
58 CV_Assert(B.get_axis_size(-2) == result.get_axis_size(-1)); in assertGEMMCompatiblity()
60 CV_Assert(A.get_axis_size(-1) == result.get_axis_size(-2)); in assertGEMMCompatiblity()
61 CV_Assert(A.get_axis_size(-2) == B.get_axis_size(-2)); in assertGEMMCompatiblity()
62 CV_Assert(B.get_axis_size(-1) == result.get_axis_size(-1)); in assertGEMMCompatiblity()
64 CV_Assert(A.get_axis_size(-1) == result.get_axis_size(-2)); in assertGEMMCompatiblity()
[all …]
H A Dtensor.hpp138 size_type get_axis_size(int axis) const noexcept { in get_axis_size() function in cv::dnn::cuda4dnn::csl::Tensor
243 new_sizes[i] = tensor.get_axis_size(i); in resize_as()
333 new_sizes[i] = tensor.get_axis_size(i); in reshape_as()
448 shape[i] = tensor.get_axis_size(i);
649 new_sizes[i] = tensor.get_axis_size(i); in reshape_as()
813 shape[i] = other.get_axis_size(i);
819 shape[i] = tensor.get_axis_size(i);
1168 if (x.get_axis_size(i) != y.get_axis_size(i)) in is_shape_same()
1184 if (x.get_axis_size(i) != y.get_axis_size(i) && in is_shape_compatible()
1185 x.get_axis_size(i) != 1 && y.get_axis_size(i) != 1) in is_shape_compatible()
[all …]
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda/
H A Ddetection_output.cu722 CV_Assert(count.get_axis_size(0) == batch_size); in findTopK()
730 const auto num_priors = scores.get_axis_size(2); in findTopK()
757 CV_Assert(count.get_axis_size(0) == batch_size); in box_collect()
793 CV_Assert(count.get_axis_size(0) == batch_size); in blockwise_class_nms()
843 auto keepTopK = kept_indices.get_axis_size(1); in nms_collect()
845 auto num_classes = indices.get_axis_size(1); in nms_collect()
849 auto classwise_topK = indices.get_axis_size(2); in nms_collect()
850 auto num_priors = scores.get_axis_size(2); in nms_collect()
882 auto keepTopK = kept_indices.get_axis_size(1); in consolidate_detections()
884 auto num_classes = scores.get_axis_size(1); in consolidate_detections()
[all …]
H A Dcrop_and_resize.cu138 CV_Assert(input.get_axis_size(0) == 1); /* batch not supported */ in crop_and_resize()
139 CV_Assert(input.get_axis_size(1) == output.get_axis_size(1)); in crop_and_resize()
141 auto out_height = output.get_axis_size(-2); in crop_and_resize()
142 auto out_width = output.get_axis_size(-1); in crop_and_resize()
144 auto in_height = input.get_axis_size(-2); in crop_and_resize()
145 auto in_width = input.get_axis_size(-1); in crop_and_resize()
147 auto num_channels = input.get_axis_size(1); in crop_and_resize()
H A Droi_pooling.cu149 CV_Assert(input.get_axis_size(1) == output.get_axis_size(1)); in roi_pooling()
151 size_type num_channels = output.get_axis_size(1); in roi_pooling()
153 size_type pooled_height = output.get_axis_size(2); in roi_pooling()
154 size_type pooled_width = output.get_axis_size(3); in roi_pooling()
156 size_type in_height = input.get_axis_size(2); in roi_pooling()
157 size_type in_width = input.get_axis_size(3); in roi_pooling()
H A Dresize.cu176 auto out_height = output.get_axis_size(-2); in resize_nn()
177 auto out_width = output.get_axis_size(-1); in resize_nn()
179 auto in_height = input.get_axis_size(-2); in resize_nn()
180 auto in_width = input.get_axis_size(-1); in resize_nn()
218 auto out_height = output.get_axis_size(-2); in resize_bilinear()
219 auto out_width = output.get_axis_size(-1); in resize_bilinear()
221 auto in_height = input.get_axis_size(-2); in resize_bilinear()
222 auto in_width = input.get_axis_size(-1); in resize_bilinear()
H A Dgrid_nms.cu402 const auto batch_size = indices.get_axis_size(0); in grid_nms()
403 CV_Assert(count.get_axis_size(0) == batch_size); in grid_nms()
404 CV_Assert(bboxes.get_axis_size(0) == batch_size); in grid_nms()
406 const auto num_classes = indices.get_axis_size(1); in grid_nms()
407 CV_Assert(count.get_axis_size(1) == num_classes); in grid_nms()
408 CV_Assert(bboxes.get_axis_size(1) == num_classes); in grid_nms()
410 const auto topK = indices.get_axis_size(2); in grid_nms()
411 CV_Assert(bboxes.get_axis_size(2) == topK); in grid_nms()
413 CV_Assert(bboxes.get_axis_size(3) == 4); in grid_nms()
H A Dmax_unpooling.cu210 CV_Assert(input.get_axis_size(1) == output.get_axis_size(1)); in max_pooling_with_indices()
220 in_spatial_dims[i] = input.get_axis_size(2 + i); in max_pooling_with_indices()
221 out_spatial_dims[i] = output.get_axis_size(2 + i); in max_pooling_with_indices()
225 std::size_t channels = input.get_axis_size(1); in max_pooling_with_indices()
288 CV_Assert(input.get_axis_size(1) == output.get_axis_size(1)); in max_unpooling()
298 in_spatial_dims[i] = input.get_axis_size(2 + i); in max_unpooling()
299 out_spatial_dims[i] = output.get_axis_size(2 + i); in max_unpooling()
306 std::size_t channels = input.get_axis_size(1); in max_unpooling()
H A Dshortcut.cu89 CV_Assert(from.get_axis_size(i) == output.get_axis_size(i)); in input_shortcut()
94 auto c_input = input.get_axis_size(1); in input_shortcut()
95 auto c_from = from.get_axis_size(1); in input_shortcut()
H A Dconcat.cu101 CV_Assert(output_axis_offset < output.get_axis_size(axis)); in concat()
128 std::size_t input_axis_size = input.get_axis_size(axis); in concat()
129 std::size_t output_axis_size = output.get_axis_size(axis); in concat()
202 while (input.get_axis_size(0) == 1 && output.get_axis_size(0) == 1) { in concat_with_offsets()
H A Dslice.cu105 while (input.get_axis_size(0) == 1 && output.get_axis_size(0) == 1) { in slice()
H A Dpadding.cu120 while (input.get_axis_size(0) == 1 && output.get_axis_size(0) == 1) { in copy_with_reflection101()
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/primitives/
H A Dmatmul.hpp55 auto size = output.get_axis_size(i); in forward()
56 CV_Assert(input1.get_axis_size(i) == size); in forward()
57 CV_Assert(input2.get_axis_size(i) == size); in forward()
60 auto m = input1.get_axis_size(-2); in forward()
61 auto n = input1.get_axis_size(-1); in forward()
62 auto k = input2.get_axis_size(-1); in forward()
64 CV_Assert(input2.get_axis_size(-2) == n); in forward()
65 CV_Assert(output.get_axis_size(-2) == m); in forward()
66 CV_Assert(output.get_axis_size(-1) == k); in forward()
H A Dreorg.hpp42 input.get_axis_size(0), in forward()
43 input.get_axis_size(1) * input.get_axis_size(2) / (stride * stride), in forward()
45 input.get_axis_size(3), in forward()
H A Dresize.hpp62 auto out_height = output.get_axis_size(-2), out_width = output.get_axis_size(-1); in forward()
63 auto in_height = input.get_axis_size(-2), in_width = input.get_axis_size(-1); in forward()
H A Dshuffle_channel.hpp51 input.get_axis_size(0), in forward()
53 input.get_axis_size(1) / group, in forward()
54 input.get_axis_size(2) * input.get_axis_size(3) in forward()
H A Dpadding.hpp68 CV_Assert(input.get_axis_size(i) == output.get_axis_size(i)); in forward()
70 CV_Assert(input.get_axis_size(i) == dstRanges[i].size()); in forward()
94 ranges[i] = { 0, input.get_axis_size(i) }; in forward()
H A Dscale_shift.hpp100 parameter_batch_size = weights.get_axis_size(0); in forward()
101 CV_Assert(parameter_batch_size == input.get_axis_size(0)); in forward()
116 parameter_batch_size = bias.get_axis_size(0); in forward()
117 CV_Assert(parameter_batch_size == input.get_axis_size(0)); in forward()
H A Dinner_product.hpp41 CV_Assert(weightsTensor.get_axis_size(-2) == biasTensor.size());
61 CV_Assert(input_size == weightsTensor.get_axis_size(-1)); in forward()
64 CV_Assert(output_size == weightsTensor.get_axis_size(-2)); in forward()
H A Dshortcut.hpp53 CV_Assert(from.get_axis_size(i) == output.get_axis_size(i)); in forward()
H A Dregion.hpp105 auto rows = input.get_axis_size(1); in forward()
106 auto cols = input.get_axis_size(2); in forward()
124 for (int i = 0; i < input.get_axis_size(0); i++) { in forward()
H A Dconcat.hpp64 output_concat_axis_offset += input.get_axis_size(concat_axis); in forward()
77 output_axis_offset += input.get_axis_size(concat_axis); in forward()
H A Ddetection_output.hpp149 auto batch_size = locations.get_axis_size(0); in forward()
H A Dactivation.hpp107 CV_Assert(input.get_axis_size(1) == slopeTensor.size()); in forward()