• Home
  • Raw
  • Download

Lines Matching refs:input2_id

33   const uint32_t input2_id = node->inputs[1];  in create_maximum_operator()  local
34 assert(input2_id != XNN_INVALID_VALUE_ID); in create_maximum_operator()
35 assert(input2_id < num_values); in create_maximum_operator()
61 opdata->shape2.num_dims = values[input2_id].shape.num_dims; in create_maximum_operator()
64 assert(values[input2_id].layout == xnn_layout_type_nchw); in create_maximum_operator()
70 opdata->shape2.dim[0] = values[input2_id].shape.dim[0]; in create_maximum_operator()
71 opdata->shape2.dim[1] = values[input2_id].shape.dim[values[input2_id].shape.num_dims - 1]; in create_maximum_operator()
73 …memcpy(&opdata->shape2.dim[2], &values[input2_id].shape.dim[1], (values[input2_id].shape.num_dims … in create_maximum_operator()
78 assert(values[input2_id].layout == xnn_layout_type_nhwc); in create_maximum_operator()
80 …memcpy(opdata->shape2.dim, values[input2_id].shape.dim, values[input2_id].shape.num_dims * sizeof(… in create_maximum_operator()
83 opdata->inputs[1] = input2_id; in create_maximum_operator()
99 const uint32_t input2_id = opdata->inputs[1]; in setup_maximum_operator() local
100 assert(input2_id != XNN_INVALID_VALUE_ID); in setup_maximum_operator()
101 assert(input2_id < num_blobs); in setup_maximum_operator()
111 const struct xnn_blob* input2_blob = blobs + input2_id; in setup_maximum_operator()
148 uint32_t input2_id, in xnn_define_maximum2() argument
180 xnn_node_type_maximum2, input2_id, subgraph->num_values, 2)) != xnn_status_success) { in xnn_define_maximum2()
184 const struct xnn_value* input2_value = &subgraph->values[input2_id]; in xnn_define_maximum2()
185 …status = xnn_subgraph_check_nth_input_type_dense(xnn_node_type_maximum2, input2_id, input2_value, … in xnn_define_maximum2()
196 xnn_node_type_to_string(xnn_node_type_maximum2), input2_id, in xnn_define_maximum2()
232 node->inputs[1] = input2_id; in xnn_define_maximum2()