Home
last modified time | relevance | path

Searched refs:num_blobs (Results 1 – 25 of 41) sorted by relevance

12

/external/XNNPACK/src/subgraph/
Deven-split.c250 const uint32_t num_blobs, in setup_even_split_operator_helper() argument
264 assert(output_id < num_blobs); in setup_even_split_operator_helper()
297 size_t num_blobs, in setup_even_split2_operator() argument
302 assert(input_id < num_blobs); in setup_even_split2_operator()
311 …status = setup_even_split_operator_helper(blobs, num_blobs, opdata, 0, channels, input_data, threa… in setup_even_split2_operator()
315 …status = setup_even_split_operator_helper(blobs, num_blobs, opdata, 1, channels, input_data, threa… in setup_even_split2_operator()
326 size_t num_blobs, in setup_even_split3_operator() argument
332 assert(input_id < num_blobs); in setup_even_split3_operator()
341 …status = setup_even_split_operator_helper(blobs, num_blobs, opdata, 0, channels, input_data, threa… in setup_even_split3_operator()
345 …status = setup_even_split_operator_helper(blobs, num_blobs, opdata, 1, channels, input_data, threa… in setup_even_split3_operator()
[all …]
Dconcatenate.c276 size_t num_blobs, in setup_concatenate2_operator() argument
281 assert(input1_id < num_blobs); in setup_concatenate2_operator()
285 assert(input2_id < num_blobs); in setup_concatenate2_operator()
289 assert(output_id < num_blobs); in setup_concatenate2_operator()
315 size_t num_blobs, in setup_concatenate3_operator() argument
320 assert(input1_id < num_blobs); in setup_concatenate3_operator()
324 assert(input2_id < num_blobs); in setup_concatenate3_operator()
328 assert(input3_id < num_blobs); in setup_concatenate3_operator()
332 assert(output_id < num_blobs); in setup_concatenate3_operator()
366 size_t num_blobs, in setup_concatenate4_operator() argument
[all …]
Dunpooling-2d.c68 size_t num_blobs, in setup_unpooling_operator() argument
73 assert(input_value_id < num_blobs); in setup_unpooling_operator()
77 assert(input_index_id < num_blobs); in setup_unpooling_operator()
81 assert(output_id < num_blobs); in setup_unpooling_operator()
Dargmax-pooling-2d.c69 size_t num_blobs, in setup_argmax_pooling_operator() argument
74 assert(input_id < num_blobs); in setup_argmax_pooling_operator()
78 assert(output_value_id < num_blobs); in setup_argmax_pooling_operator()
82 assert(output_index_id < num_blobs); in setup_argmax_pooling_operator()
Dsquare-root.c71 size_t num_blobs, in setup_square_root_operator() argument
76 assert(input_id < num_blobs); in setup_square_root_operator()
80 assert(output_id < num_blobs); in setup_square_root_operator()
Dceiling.c69 size_t num_blobs, in setup_ceiling_operator() argument
74 assert(input_id < num_blobs); in setup_ceiling_operator()
78 assert(output_id < num_blobs); in setup_ceiling_operator()
Dsquare.c69 size_t num_blobs, in setup_square_operator() argument
74 assert(input_id < num_blobs); in setup_square_operator()
78 assert(output_id < num_blobs); in setup_square_operator()
Dfloor.c69 size_t num_blobs, in setup_floor_operator() argument
74 assert(input_id < num_blobs); in setup_floor_operator()
78 assert(output_id < num_blobs); in setup_floor_operator()
Dsoftmax.c70 size_t num_blobs, in setup_softmax_operator() argument
75 assert(input_id < num_blobs); in setup_softmax_operator()
79 assert(output_id < num_blobs); in setup_softmax_operator()
Dabs.c68 size_t num_blobs, in setup_abs_operator() argument
73 assert(input_id < num_blobs); in setup_abs_operator()
77 assert(output_id < num_blobs); in setup_abs_operator()
Dhardswish.c71 size_t num_blobs, in setup_hardswish_operator() argument
76 assert(input_id < num_blobs); in setup_hardswish_operator()
80 assert(output_id < num_blobs); in setup_hardswish_operator()
Dbankers-rounding.c69 size_t num_blobs, in setup_bankers_rounding_operator() argument
74 assert(input_id < num_blobs); in setup_bankers_rounding_operator()
78 assert(output_id < num_blobs); in setup_bankers_rounding_operator()
Dnegate.c69 size_t num_blobs, in setup_negate_operator() argument
74 assert(input_id < num_blobs); in setup_negate_operator()
78 assert(output_id < num_blobs); in setup_negate_operator()
Dminimum2.c92 size_t num_blobs, in setup_minimum_operator() argument
97 assert(input1_id < num_blobs); in setup_minimum_operator()
101 assert(input2_id < num_blobs); in setup_minimum_operator()
105 assert(output_id < num_blobs); in setup_minimum_operator()
Dmaximum2.c92 size_t num_blobs, in setup_maximum_operator() argument
97 assert(input1_id < num_blobs); in setup_maximum_operator()
101 assert(input2_id < num_blobs); in setup_maximum_operator()
105 assert(output_id < num_blobs); in setup_maximum_operator()
Dsquared-difference.c92 size_t num_blobs, in setup_squared_difference_operator() argument
97 assert(input1_id < num_blobs); in setup_squared_difference_operator()
101 assert(input2_id < num_blobs); in setup_squared_difference_operator()
105 assert(output_id < num_blobs); in setup_squared_difference_operator()
Ddivide.c95 size_t num_blobs, in setup_divide_operator() argument
100 assert(input1_id < num_blobs); in setup_divide_operator()
104 assert(input2_id < num_blobs); in setup_divide_operator()
108 assert(output_id < num_blobs); in setup_divide_operator()
Dprelu.c76 size_t num_blobs, in setup_prelu_operator() argument
81 assert(input_id < num_blobs); in setup_prelu_operator()
85 assert(output_id < num_blobs); in setup_prelu_operator()
Delu.c85 size_t num_blobs, in setup_elu_operator() argument
90 assert(input_id < num_blobs); in setup_elu_operator()
94 assert(output_id < num_blobs); in setup_elu_operator()
Dsigmoid.c99 size_t num_blobs, in setup_sigmoid_operator() argument
104 assert(input_id < num_blobs); in setup_sigmoid_operator()
108 assert(output_id < num_blobs); in setup_sigmoid_operator()
Dstatic-reshape.c80 size_t num_blobs, in setup_copy_operator() argument
85 assert(input_id < num_blobs); in setup_copy_operator()
89 assert(output_id < num_blobs); in setup_copy_operator()
Dstatic-transpose.c70 size_t num_blobs, in setup_transpose_operator() argument
75 assert(input_id < num_blobs); in setup_transpose_operator()
79 assert(output_id < num_blobs); in setup_transpose_operator()
Dadd2.c131 size_t num_blobs, in setup_add_operator() argument
136 assert(input1_id < num_blobs); in setup_add_operator()
140 assert(input2_id < num_blobs); in setup_add_operator()
144 assert(output_id < num_blobs); in setup_add_operator()
Dsubtract.c131 size_t num_blobs, in setup_subtract_operator() argument
136 assert(input1_id < num_blobs); in setup_subtract_operator()
140 assert(input2_id < num_blobs); in setup_subtract_operator()
144 assert(output_id < num_blobs); in setup_subtract_operator()
/external/XNNPACK/test/
Dworkspace.cc159 for (size_t i = 0; i < runtime1->num_blobs; i++) { in TEST()
169 for (size_t i = 0; i < runtime2->num_blobs; i++) { in TEST()
216 ASSERT_EQ(runtime1->num_blobs, runtime2->num_blobs); in TEST()
217 for (size_t i = 0; i < runtime1->num_blobs; i++) { in TEST()
276 for (size_t i = 0; i < runtime1->num_blobs; i++) { in TEST()
283 for (size_t i = 0; i < runtime2->num_blobs; i++) { in TEST()

12