Home
last modified time | relevance | path

Searched refs:output_id (Results 1 – 25 of 52) sorted by relevance

123

/external/XNNPACK/src/subgraph/
Dsoftmax.c19 uint32_t output_id, in xnn_define_softmax() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_softmax()
38 xnn_node_type_to_string(xnn_node_type_softmax), output_id); in xnn_define_softmax()
51 node->outputs[0] = output_id; in xnn_define_softmax()
Dfloor.c19 uint32_t output_id, in xnn_define_floor() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_floor()
38 xnn_node_type_to_string(xnn_node_type_floor), output_id); in xnn_define_floor()
51 node->outputs[0] = output_id; in xnn_define_floor()
Dsquare-root.c19 uint32_t output_id, in xnn_define_square_root() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_square_root()
38 xnn_node_type_to_string(xnn_node_type_square_root), output_id); in xnn_define_square_root()
51 node->outputs[0] = output_id; in xnn_define_square_root()
Dbankers-rounding.c19 uint32_t output_id, in xnn_define_bankers_rounding() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_bankers_rounding()
38 xnn_node_type_to_string(xnn_node_type_bankers_rounding), output_id); in xnn_define_bankers_rounding()
51 node->outputs[0] = output_id; in xnn_define_bankers_rounding()
Dceiling.c19 uint32_t output_id, in xnn_define_ceiling() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_ceiling()
38 xnn_node_type_to_string(xnn_node_type_ceiling), output_id); in xnn_define_ceiling()
51 node->outputs[0] = output_id; in xnn_define_ceiling()
Dnegate.c19 uint32_t output_id, in xnn_define_negate() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_negate()
38 xnn_node_type_to_string(xnn_node_type_negate), output_id); in xnn_define_negate()
51 node->outputs[0] = output_id; in xnn_define_negate()
Dsigmoid.c19 uint32_t output_id, in xnn_define_sigmoid() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_sigmoid()
38 xnn_node_type_to_string(xnn_node_type_sigmoid), output_id); in xnn_define_sigmoid()
51 node->outputs[0] = output_id; in xnn_define_sigmoid()
Dabs.c19 uint32_t output_id, in xnn_define_abs() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_abs()
38 xnn_node_type_to_string(xnn_node_type_abs), output_id); in xnn_define_abs()
51 node->outputs[0] = output_id; in xnn_define_abs()
Dsquare.c19 uint32_t output_id, in xnn_define_square() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_square()
38 xnn_node_type_to_string(xnn_node_type_square), output_id); in xnn_define_square()
51 node->outputs[0] = output_id; in xnn_define_square()
Dhardswish.c19 uint32_t output_id, in xnn_define_hardswish() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_hardswish()
38 xnn_node_type_to_string(xnn_node_type_hardswish), output_id); in xnn_define_hardswish()
51 node->outputs[0] = output_id; in xnn_define_hardswish()
Dclamp.c21 uint32_t output_id, in xnn_define_clamp() argument
37 if (output_id >= subgraph->num_values) { in xnn_define_clamp()
40 xnn_node_type_to_string(xnn_node_type_clamp), output_id); in xnn_define_clamp()
55 node->outputs[0] = output_id; in xnn_define_clamp()
Dprelu.c20 uint32_t output_id, in xnn_define_prelu() argument
43 if (output_id >= subgraph->num_values) { in xnn_define_prelu()
46 xnn_node_type_to_string(xnn_node_type_prelu), output_id); in xnn_define_prelu()
60 node->outputs[0] = output_id; in xnn_define_prelu()
Ddepth-to-space.c18 uint32_t output_id, in xnn_define_depth_to_space() argument
35 if (output_id >= subgraph->num_values) { in xnn_define_depth_to_space()
38 xnn_node_type_to_string(xnn_node_type_depth_to_space), output_id); in xnn_define_depth_to_space()
58 node->outputs[0] = output_id; in xnn_define_depth_to_space()
Dsquared-difference.c20 uint32_t output_id, in xnn_define_squared_difference() argument
43 if (output_id >= subgraph->num_values) { in xnn_define_squared_difference()
46 xnn_node_type_to_string(xnn_node_type_squared_difference), output_id); in xnn_define_squared_difference()
60 node->outputs[0] = output_id; in xnn_define_squared_difference()
Dleaky-relu.c20 uint32_t output_id, in xnn_define_leaky_relu() argument
44 if (output_id >= subgraph->num_values) { in xnn_define_leaky_relu()
47 xnn_node_type_to_string(xnn_node_type_leaky_relu), output_id); in xnn_define_leaky_relu()
61 node->outputs[0] = output_id; in xnn_define_leaky_relu()
Dmaximum2.c20 uint32_t output_id, in xnn_define_maximum2() argument
43 if (output_id >= subgraph->num_values) { in xnn_define_maximum2()
46 xnn_node_type_to_string(xnn_node_type_maximum2), output_id); in xnn_define_maximum2()
60 node->outputs[0] = output_id; in xnn_define_maximum2()
Dminimum2.c20 uint32_t output_id, in xnn_define_minimum2() argument
43 if (output_id >= subgraph->num_values) { in xnn_define_minimum2()
46 xnn_node_type_to_string(xnn_node_type_minimum2), output_id); in xnn_define_minimum2()
60 node->outputs[0] = output_id; in xnn_define_minimum2()
Delu.c20 uint32_t output_id, in xnn_define_elu() argument
43 if (output_id >= subgraph->num_values) { in xnn_define_elu()
46 xnn_node_type_to_string(xnn_node_type_elu), output_id); in xnn_define_elu()
60 node->outputs[0] = output_id; in xnn_define_elu()
Dstatic-reshape.c22 uint32_t output_id, in xnn_define_static_reshape() argument
38 if (output_id >= subgraph->num_values) { in xnn_define_static_reshape()
41 xnn_node_type_to_string(xnn_node_type_static_reshape), output_id); in xnn_define_static_reshape()
64 node->outputs[0] = output_id; in xnn_define_static_reshape()
Dstatic-constant-pad.c25 uint32_t output_id, in xnn_define_static_constant_pad() argument
41 if (output_id >= subgraph->num_values) { in xnn_define_static_constant_pad()
44 xnn_node_type_to_string(xnn_node_type_static_constant_pad), output_id); in xnn_define_static_constant_pad()
62 node->outputs[0] = output_id; in xnn_define_static_constant_pad()
Dglobal-average-pooling-2d.c21 uint32_t output_id, in xnn_define_global_average_pooling_2d() argument
58 if (output_id >= subgraph->num_values) { in xnn_define_global_average_pooling_2d()
61 xnn_node_type_to_string(xnn_node_type_global_average_pooling_2d), output_id); in xnn_define_global_average_pooling_2d()
76 node->outputs[0] = output_id; in xnn_define_global_average_pooling_2d()
Ddivide.c22 uint32_t output_id, in xnn_define_divide() argument
66 if (output_id >= subgraph->num_values) { in xnn_define_divide()
69 xnn_node_type_to_string(xnn_node_type_divide), output_id); in xnn_define_divide()
85 node->outputs[0] = output_id; in xnn_define_divide()
Dmultiply2.c22 uint32_t output_id, in xnn_define_multiply2() argument
66 if (output_id >= subgraph->num_values) { in xnn_define_multiply2()
69 xnn_node_type_to_string(xnn_node_type_multiply2), output_id); in xnn_define_multiply2()
85 node->outputs[0] = output_id; in xnn_define_multiply2()
/external/XNNPACK/test/
Dsubgraph-tester.h82 uint32_t output_id) in add_conv() argument
91 output_id, 0 /* flags */); in add_conv()
104 uint32_t filter_id, uint32_t bias_id, uint32_t output_id) in add_depthwise_conv() argument
113 output_id, 0 /* flags */); in add_depthwise_conv()
119 inline SubgraphTester& add_addition(uint32_t input_id1, uint32_t input_id2, uint32_t output_id) in add_addition() argument
124 input_id2, output_id, 0 /* flags */); in add_addition()
130 inline SubgraphTester& add_global_average_pooling(uint32_t input_id, uint32_t output_id) in add_global_average_pooling() argument
134 std::numeric_limits<float>::infinity(), input_id, output_id, 0 /* flags */); in add_global_average_pooling()
/external/XNNPACK/include/
Dxnnpack.h257 uint32_t output_id,
312 uint32_t output_id,
365 uint32_t output_id,
382 uint32_t output_id,
404 uint32_t output_id,
445 uint32_t output_id,
495 uint32_t output_id, uint32_t flags);
539 uint32_t output_id,
599 uint32_t output_id,
627 uint32_t output_id,
[all …]

123