Home
last modified time | relevance | path

Searched refs:input_id (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/python/framework/
Dauto_control_deps.py286 input_id = ops.tensor_id(inp)
300 if input_id in last_write_to_resource:
302 switch_op._add_control_input(last_write_to_resource[input_id])
304 last_write_to_resource[input_id] = new_merge[0].op
305 if input_id in merge_for_resource:
306 merge_for_resource[input_id]._add_control_input(new_merge[0].op)
402 input_id = ops.tensor_id(inp)
403 if input_id in last_write_to_resource:
404 last_write_to_resource[input_id] = op
413 input_id = ops.tensor_id(inp)
[all …]
/external/XNNPACK/src/subgraph/
Dstatic-constant-pad.c24 uint32_t input_id, in xnn_define_static_constant_pad() argument
34 if (input_id >= subgraph->num_values) { in xnn_define_static_constant_pad()
37 xnn_node_type_to_string(xnn_node_type_static_constant_pad), input_id); in xnn_define_static_constant_pad()
53 const size_t num_dims = subgraph->values[input_id].shape.num_dims; in xnn_define_static_constant_pad()
60 node->inputs[0] = input_id; in xnn_define_static_constant_pad()
Dfloor.c18 uint32_t input_id, in xnn_define_floor() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_floor()
31 xnn_node_type_to_string(xnn_node_type_floor), input_id); in xnn_define_floor()
49 node->inputs[0] = input_id; in xnn_define_floor()
Dabs.c18 uint32_t input_id, in xnn_define_abs() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_abs()
31 xnn_node_type_to_string(xnn_node_type_abs), input_id); in xnn_define_abs()
49 node->inputs[0] = input_id; in xnn_define_abs()
Dhardswish.c18 uint32_t input_id, in xnn_define_hardswish() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_hardswish()
31 xnn_node_type_to_string(xnn_node_type_hardswish), input_id); in xnn_define_hardswish()
49 node->inputs[0] = input_id; in xnn_define_hardswish()
Dbankers-rounding.c18 uint32_t input_id, in xnn_define_bankers_rounding() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_bankers_rounding()
31 xnn_node_type_to_string(xnn_node_type_bankers_rounding), input_id); in xnn_define_bankers_rounding()
49 node->inputs[0] = input_id; in xnn_define_bankers_rounding()
Dsquare-root.c18 uint32_t input_id, in xnn_define_square_root() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_square_root()
31 xnn_node_type_to_string(xnn_node_type_square_root), input_id); in xnn_define_square_root()
49 node->inputs[0] = input_id; in xnn_define_square_root()
Dsigmoid.c18 uint32_t input_id, in xnn_define_sigmoid() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_sigmoid()
31 xnn_node_type_to_string(xnn_node_type_sigmoid), input_id); in xnn_define_sigmoid()
49 node->inputs[0] = input_id; in xnn_define_sigmoid()
Dsquare.c18 uint32_t input_id, in xnn_define_square() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_square()
31 xnn_node_type_to_string(xnn_node_type_square), input_id); in xnn_define_square()
49 node->inputs[0] = input_id; in xnn_define_square()
Dnegate.c18 uint32_t input_id, in xnn_define_negate() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_negate()
31 xnn_node_type_to_string(xnn_node_type_negate), input_id); in xnn_define_negate()
49 node->inputs[0] = input_id; in xnn_define_negate()
Dsoftmax.c18 uint32_t input_id, in xnn_define_softmax() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_softmax()
31 xnn_node_type_to_string(xnn_node_type_softmax), input_id); in xnn_define_softmax()
49 node->inputs[0] = input_id; in xnn_define_softmax()
Dceiling.c18 uint32_t input_id, in xnn_define_ceiling() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_ceiling()
31 xnn_node_type_to_string(xnn_node_type_ceiling), input_id); in xnn_define_ceiling()
49 node->inputs[0] = input_id; in xnn_define_ceiling()
Dclamp.c20 uint32_t input_id, in xnn_define_clamp() argument
30 if (input_id >= subgraph->num_values) { in xnn_define_clamp()
33 xnn_node_type_to_string(xnn_node_type_clamp), input_id); in xnn_define_clamp()
53 node->inputs[0] = input_id; in xnn_define_clamp()
Ddepth-to-space.c17 uint32_t input_id, in xnn_define_depth_to_space() argument
28 if (input_id >= subgraph->num_values) { in xnn_define_depth_to_space()
31 xnn_node_type_to_string(xnn_node_type_depth_to_space), input_id); in xnn_define_depth_to_space()
56 node->inputs[0] = input_id; in xnn_define_depth_to_space()
Dleaky-relu.c19 uint32_t input_id, in xnn_define_leaky_relu() argument
37 if (input_id >= subgraph->num_values) { in xnn_define_leaky_relu()
40 xnn_node_type_to_string(xnn_node_type_leaky_relu), input_id); in xnn_define_leaky_relu()
59 node->inputs[0] = input_id; in xnn_define_leaky_relu()
Dprelu.c18 uint32_t input_id, in xnn_define_prelu() argument
29 if (input_id >= subgraph->num_values) { in xnn_define_prelu()
32 xnn_node_type_to_string(xnn_node_type_prelu), input_id); in xnn_define_prelu()
57 node->inputs[0] = input_id; in xnn_define_prelu()
Delu.c19 uint32_t input_id, in xnn_define_elu() argument
36 if (input_id >= subgraph->num_values) { in xnn_define_elu()
39 xnn_node_type_to_string(xnn_node_type_elu), input_id); in xnn_define_elu()
58 node->inputs[0] = input_id; in xnn_define_elu()
Dstatic-reshape.c21 uint32_t input_id, in xnn_define_static_reshape() argument
31 if (input_id >= subgraph->num_values) { in xnn_define_static_reshape()
34 xnn_node_type_to_string(xnn_node_type_static_reshape), input_id); in xnn_define_static_reshape()
62 node->inputs[0] = input_id; in xnn_define_static_reshape()
Dglobal-average-pooling-2d.c20 uint32_t input_id, in xnn_define_global_average_pooling_2d() argument
51 if (input_id >= subgraph->num_values) { in xnn_define_global_average_pooling_2d()
54 xnn_node_type_to_string(xnn_node_type_global_average_pooling_2d), input_id); in xnn_define_global_average_pooling_2d()
74 node->inputs[0] = input_id; in xnn_define_global_average_pooling_2d()
Dstatic-resize-bilinear-2d.c20 uint32_t input_id, in xnn_define_static_resize_bilinear_2d() argument
62 if (input_id >= subgraph->num_values) { in xnn_define_static_resize_bilinear_2d()
65 xnn_node_type_to_string(xnn_node_type_static_resize_bilinear_2d), input_id); in xnn_define_static_resize_bilinear_2d()
86 node->inputs[0] = input_id; in xnn_define_static_resize_bilinear_2d()
Dfully-connected.c20 uint32_t input_id, in xnn_define_fully_connected() argument
53 if (input_id >= subgraph->num_values) { in xnn_define_fully_connected()
56 xnn_node_type_to_string(xnn_node_type_fully_connected), input_id); in xnn_define_fully_connected()
90 node->inputs[0] = input_id; in xnn_define_fully_connected()
Dargmax-pooling-2d.c24 uint32_t input_id, in xnn_define_argmax_pooling_2d() argument
51 if (input_id >= subgraph->num_values) { in xnn_define_argmax_pooling_2d()
54 xnn_node_type_to_string(xnn_node_type_argmax_pooling_2d), input_id); in xnn_define_argmax_pooling_2d()
85 node->inputs[0] = input_id; in xnn_define_argmax_pooling_2d()
Daverage-pooling-2d.c28 uint32_t input_id, in xnn_define_average_pooling_2d() argument
95 if (input_id >= subgraph->num_values) { in xnn_define_average_pooling_2d()
98 xnn_node_type_to_string(xnn_node_type_average_pooling_2d), input_id); in xnn_define_average_pooling_2d()
126 node->inputs[0] = input_id; in xnn_define_average_pooling_2d()
/external/XNNPACK/test/
Dsubgraph-tester.h81 uint32_t input_id, uint32_t filter_id, uint32_t bias_id, in add_conv() argument
90 std::numeric_limits<float>::infinity(), input_id, filter_id, bias_id, in add_conv()
103 uint32_t depth_multiplier, size_t input_channels, uint32_t input_id, in add_depthwise_conv() argument
112 std::numeric_limits<float>::infinity(), input_id, filter_id, bias_id, in add_depthwise_conv()
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.h254 uint32_t input_id,
309 uint32_t input_id,
362 uint32_t input_id,
381 uint32_t input_id,
403 uint32_t input_id,
444 uint32_t input_id,
493 uint32_t input_id,
538 uint32_t input_id,
567 uint32_t input_id,
804 uint32_t input_id,
[all …]

123