Lines Matching refs:graph_node
531 auto graph_node = node->xnode_union_as_XNNAdd(); in defineAddNode() local
536 remapped_ids.at(graph_node->input1_id()), in defineAddNode()
537 remapped_ids.at(graph_node->input2_id()), in defineAddNode()
538 remapped_ids.at(graph_node->output_id()), in defineAddNode()
539 graph_node->flags()); in defineAddNode()
560 auto graph_node = node->xnode_union_as_XNNMinimum(); in defineMinimumNode() local
563 remapped_ids.at(graph_node->input1_id()), in defineMinimumNode()
564 remapped_ids.at(graph_node->input2_id()), in defineMinimumNode()
565 remapped_ids.at(graph_node->output_id()), in defineMinimumNode()
566 graph_node->flags()); in defineMinimumNode()
588 auto graph_node = node->xnode_union_as_XNNSubtract(); in defineSubtractNode() local
594 remapped_ids.at(graph_node->input1_id()), in defineSubtractNode()
595 remapped_ids.at(graph_node->input2_id()), in defineSubtractNode()
596 remapped_ids.at(graph_node->output_id()), in defineSubtractNode()
597 graph_node->flags()); in defineSubtractNode()
619 auto graph_node = node->xnode_union_as_XNNMultiply(); in defineMultiplyNode() local
625 remapped_ids.at(graph_node->input1_id()), in defineMultiplyNode()
626 remapped_ids.at(graph_node->input2_id()), in defineMultiplyNode()
627 remapped_ids.at(graph_node->output_id()), in defineMultiplyNode()
628 graph_node->flags()); in defineMultiplyNode()
643 auto graph_node = node->xnode_union_as_XNNConvert(); in isQP8() local
644 auto cvt_output_id = graph_node->output_id(); in isQP8()
696 auto graph_node = node->xnode_union_as_XNNConvert(); in defineConvertNode() local
698 int32_t flags = graph_node->flags(); in defineConvertNode()
714 remapped_ids.at(graph_node->input_id()), in defineConvertNode()
715 remapped_ids.at(graph_node->output_id()), in defineConvertNode()
739 auto graph_node = node->xnode_union_as_XNNFullyConnected(); in defineFullyConnectedNode() local
745 remapped_ids.at(graph_node->input1_id()), in defineFullyConnectedNode()
746 remapped_ids.at(graph_node->filter_id()), in defineFullyConnectedNode()
747 remapped_ids.at(graph_node->bias_id()), in defineFullyConnectedNode()
748 remapped_ids.at(graph_node->output_id()), in defineFullyConnectedNode()
749 graph_node->flags()); in defineFullyConnectedNode()
773 auto graph_node = node->xnode_union_as_XNNClamp(); in defineClampNode() local
778 remapped_ids.at(graph_node->input_id()), in defineClampNode()
779 remapped_ids.at(graph_node->output_id()), in defineClampNode()
780 graph_node->flags()); in defineClampNode()
804 auto graph_node = node->xnode_union_as_XNNSoftmax(); in defineSoftmaxNode() local
807 remapped_ids.at(graph_node->input_id()), in defineSoftmaxNode()
808 remapped_ids.at(graph_node->output_id()), in defineSoftmaxNode()
809 graph_node->flags()); in defineSoftmaxNode()
832 auto graph_node = node->xnode_union_as_XNNSigmoid(); in defineSigmoidNode() local
835 remapped_ids.at(graph_node->input_id()), in defineSigmoidNode()
836 remapped_ids.at(graph_node->output_id()), in defineSigmoidNode()
837 graph_node->flags()); in defineSigmoidNode()
860 auto graph_node = node->xnode_union_as_XNNFloor(); in defineFloorNode() local
863 remapped_ids.at(graph_node->input_id()), in defineFloorNode()
864 remapped_ids.at(graph_node->output_id()), in defineFloorNode()
865 graph_node->flags()); in defineFloorNode()
883 auto graph_node = node->xnode_union_as_XNNGlobalAvgPooling2d(); in defineGlobalAvgPooling2dNode() local
889 remapped_ids.at(graph_node->input_id()), in defineGlobalAvgPooling2dNode()
890 remapped_ids.at(graph_node->output_id()), in defineGlobalAvgPooling2dNode()
891 graph_node->flags()); in defineGlobalAvgPooling2dNode()
909 auto graph_node = node->xnode_union_as_XNNAvgPooling2d(); in defineAvgPooling2dNode() local
913 graph_node->padding_top(), in defineAvgPooling2dNode()
914 graph_node->padding_right(), in defineAvgPooling2dNode()
915 graph_node->padding_bottom(), in defineAvgPooling2dNode()
916 graph_node->padding_left(), in defineAvgPooling2dNode()
917 graph_node->pooling_height(), in defineAvgPooling2dNode()
918 graph_node->pooling_width(), in defineAvgPooling2dNode()
919 graph_node->stride_height(), in defineAvgPooling2dNode()
920 graph_node->stride_width(), in defineAvgPooling2dNode()
923 remapped_ids.at(graph_node->input_id()), in defineAvgPooling2dNode()
924 remapped_ids.at(graph_node->output_id()), in defineAvgPooling2dNode()
925 graph_node->flags()); in defineAvgPooling2dNode()
948 auto graph_node = node->xnode_union_as_XNNConv2d(); in defineConv2dNode() local
952 graph_node->padding_top(), in defineConv2dNode()
953 graph_node->padding_right(), in defineConv2dNode()
954 graph_node->padding_bottom(), in defineConv2dNode()
955 graph_node->padding_left(), in defineConv2dNode()
956 graph_node->kernel_height(), in defineConv2dNode()
957 graph_node->kernel_width(), in defineConv2dNode()
958 graph_node->subsampling_height(), in defineConv2dNode()
959 graph_node->subsampling_width(), in defineConv2dNode()
960 graph_node->dilation_height(), in defineConv2dNode()
961 graph_node->dilation_width(), in defineConv2dNode()
962 graph_node->groups(), in defineConv2dNode()
963 graph_node->group_input_channels(), in defineConv2dNode()
964 graph_node->group_output_channels(), in defineConv2dNode()
967 remapped_ids.at(graph_node->input1_id()), in defineConv2dNode()
968 remapped_ids.at(graph_node->filter_id()), in defineConv2dNode()
969 remapped_ids.at(graph_node->bias_id()), in defineConv2dNode()
970 remapped_ids.at(graph_node->output_id()), in defineConv2dNode()
971 graph_node->flags()); in defineConv2dNode()
994 auto graph_node = node->xnode_union_as_XNNMaxPooling2d(); in defineMaxPooling2dNode() local
998 graph_node->padding_top(), in defineMaxPooling2dNode()
999 graph_node->padding_right(), in defineMaxPooling2dNode()
1000 graph_node->padding_bottom(), in defineMaxPooling2dNode()
1001 graph_node->padding_left(), in defineMaxPooling2dNode()
1002 graph_node->pooling_height(), in defineMaxPooling2dNode()
1003 graph_node->pooling_width(), in defineMaxPooling2dNode()
1004 graph_node->stride_height(), in defineMaxPooling2dNode()
1005 graph_node->stride_width(), in defineMaxPooling2dNode()
1006 graph_node->dilation_height(), in defineMaxPooling2dNode()
1007 graph_node->dilation_width(), in defineMaxPooling2dNode()
1010 remapped_ids.at(graph_node->input_id()), in defineMaxPooling2dNode()
1011 remapped_ids.at(graph_node->output_id()), in defineMaxPooling2dNode()
1012 graph_node->flags()); in defineMaxPooling2dNode()
1034 auto graph_node = node->xnode_union_as_XNNDiv(); in defineDivNode() local
1040 remapped_ids.at(graph_node->input1_id()), in defineDivNode()
1041 remapped_ids.at(graph_node->input2_id()), in defineDivNode()
1042 remapped_ids.at(graph_node->output_id()), in defineDivNode()
1043 graph_node->flags()); in defineDivNode()
1066 auto graph_node = node->xnode_union_as_XNNStaticTranspose(); in defineStaticTransposeNode() local
1069 std::vector<size_t> dims_data = flatbufferDimsToVector(graph_node->perm()); in defineStaticTransposeNode()
1072 graph_node->num_dims(), in defineStaticTransposeNode()
1074 remapped_ids.at(graph_node->input_id()), in defineStaticTransposeNode()
1075 remapped_ids.at(graph_node->output_id()), in defineStaticTransposeNode()
1076 graph_node->flags()); in defineStaticTransposeNode()
1099 const fb_xnnpack::XNNStaticResizeBilinear2D* graph_node = in defineStaticResizeBilinear2DNode() local
1104 graph_node->new_height(), in defineStaticResizeBilinear2DNode()
1105 graph_node->new_width(), in defineStaticResizeBilinear2DNode()
1106 remapped_ids.at(graph_node->input_id()), in defineStaticResizeBilinear2DNode()
1107 remapped_ids.at(graph_node->output_id()), in defineStaticResizeBilinear2DNode()
1108 graph_node->flags()); in defineStaticResizeBilinear2DNode()
1131 const fb_xnnpack::XNNStaticConstantPad* graph_node = in defineStaticConstantPadNode() local
1135 flatbufferDimsToVector(graph_node->pre_paddings()); in defineStaticConstantPadNode()
1137 flatbufferDimsToVector(graph_node->post_paddings()); in defineStaticConstantPadNode()
1143 graph_node->padding_value(), in defineStaticConstantPadNode()
1144 remapped_ids.at(graph_node->input_id()), in defineStaticConstantPadNode()
1145 remapped_ids.at(graph_node->output_id()), in defineStaticConstantPadNode()
1146 graph_node->flags()); in defineStaticConstantPadNode()
1169 auto graph_node = node->xnode_union_as_XNNDepthwiseConv2d(); in defineDepthwiseConv2dNode() local
1173 graph_node->padding_top(), in defineDepthwiseConv2dNode()
1174 graph_node->padding_right(), in defineDepthwiseConv2dNode()
1175 graph_node->padding_bottom(), in defineDepthwiseConv2dNode()
1176 graph_node->padding_left(), in defineDepthwiseConv2dNode()
1177 graph_node->kernel_height(), in defineDepthwiseConv2dNode()
1178 graph_node->kernel_width(), in defineDepthwiseConv2dNode()
1179 graph_node->subsampling_height(), in defineDepthwiseConv2dNode()
1180 graph_node->subsampling_width(), in defineDepthwiseConv2dNode()
1181 graph_node->dilation_height(), in defineDepthwiseConv2dNode()
1182 graph_node->dilation_width(), in defineDepthwiseConv2dNode()
1183 graph_node->group_output_channels() / in defineDepthwiseConv2dNode()
1184 graph_node->group_input_channels(), // depth_multiplier in defineDepthwiseConv2dNode()
1185 graph_node->groups(), // input_channels = groups for depthwise conv in defineDepthwiseConv2dNode()
1188 remapped_ids.at(graph_node->input1_id()), in defineDepthwiseConv2dNode()
1189 remapped_ids.at(graph_node->filter_id()), in defineDepthwiseConv2dNode()
1190 remapped_ids.at(graph_node->bias_id()), in defineDepthwiseConv2dNode()
1191 remapped_ids.at(graph_node->output_id()), in defineDepthwiseConv2dNode()
1192 graph_node->flags()); in defineDepthwiseConv2dNode()
1211 auto graph_node = node->xnode_union_as_XNNStaticReshape(); in defineStaticReshapeNode() local
1215 flatbufferDimsToVector(graph_node->new_shape()); in defineStaticReshapeNode()
1218 graph_node->num_dims(), in defineStaticReshapeNode()
1220 remapped_ids.at(graph_node->input_id()), in defineStaticReshapeNode()
1221 remapped_ids.at(graph_node->output_id()), in defineStaticReshapeNode()
1222 graph_node->flags()); in defineStaticReshapeNode()
1245 auto graph_node = node->xnode_union_as_XNNArgMaxPooling2d(); in defineArgMaxPooling2dNode() local
1249 graph_node->padding_top(), in defineArgMaxPooling2dNode()
1250 graph_node->padding_right(), in defineArgMaxPooling2dNode()
1251 graph_node->padding_bottom(), in defineArgMaxPooling2dNode()
1252 graph_node->padding_left(), in defineArgMaxPooling2dNode()
1253 graph_node->pooling_height(), in defineArgMaxPooling2dNode()
1254 graph_node->pooling_width(), in defineArgMaxPooling2dNode()
1255 remapped_ids.at(graph_node->input_id()), in defineArgMaxPooling2dNode()
1256 remapped_ids.at(graph_node->output_value_id()), in defineArgMaxPooling2dNode()
1257 remapped_ids.at(graph_node->output_index_id()), in defineArgMaxPooling2dNode()
1258 graph_node->flags()); in defineArgMaxPooling2dNode()
1282 auto graph_node = node->xnode_union_as_XNNSquareRoot(); in defineSquareRootNode() local
1286 remapped_ids.at(graph_node->input_id()), in defineSquareRootNode()
1287 remapped_ids.at(graph_node->output_id()), in defineSquareRootNode()
1288 graph_node->flags()); in defineSquareRootNode()
1312 auto graph_node = node->xnode_union_as_XNNCeiling(); in defineCeilingNode() local
1316 remapped_ids.at(graph_node->input_id()), in defineCeilingNode()
1317 remapped_ids.at(graph_node->output_id()), in defineCeilingNode()
1318 graph_node->flags()); in defineCeilingNode()
1342 auto graph_node = node->xnode_union_as_XNNHardswish(); in defineHardswishNode() local
1346 remapped_ids.at(graph_node->input_id()), in defineHardswishNode()
1347 remapped_ids.at(graph_node->output_id()), in defineHardswishNode()
1348 graph_node->flags()); in defineHardswishNode()
1372 auto graph_node = node->xnode_union_as_XNNLeakyReLU(); in defineLeakyReLUNode() local
1376 graph_node->negative_slope(), in defineLeakyReLUNode()
1377 remapped_ids.at(graph_node->input_id()), in defineLeakyReLUNode()
1378 remapped_ids.at(graph_node->output_id()), in defineLeakyReLUNode()
1379 graph_node->flags()); in defineLeakyReLUNode()
1403 auto graph_node = node->xnode_union_as_XNNMaximum(); in defineMaximumNode() local
1407 remapped_ids.at(graph_node->input1_id()), in defineMaximumNode()
1408 remapped_ids.at(graph_node->input2_id()), in defineMaximumNode()
1409 remapped_ids.at(graph_node->output_id()), in defineMaximumNode()
1410 graph_node->flags()); in defineMaximumNode()
1433 auto graph_node = node->xnode_union_as_XNNNegate(); in defineNegateNode() local
1437 remapped_ids.at(graph_node->input_id()), in defineNegateNode()
1438 remapped_ids.at(graph_node->output_id()), in defineNegateNode()
1439 graph_node->flags()); in defineNegateNode()
1462 auto graph_node = node->xnode_union_as_XNNSquare(); in defineSquareNode() local
1466 remapped_ids.at(graph_node->input_id()), in defineSquareNode()
1467 remapped_ids.at(graph_node->output_id()), in defineSquareNode()
1468 graph_node->flags()); in defineSquareNode()
1491 auto graph_node = node->xnode_union_as_XNNELU(); in defineELUNode() local
1495 graph_node->alpha(), in defineELUNode()
1496 remapped_ids.at(graph_node->input_id()), in defineELUNode()
1497 remapped_ids.at(graph_node->output_id()), in defineELUNode()
1498 graph_node->flags()); in defineELUNode()
1521 auto graph_node = node->xnode_union_as_XNNAbs(); in defineAbsNode() local
1525 remapped_ids.at(graph_node->input_id()), in defineAbsNode()
1526 remapped_ids.at(graph_node->output_id()), in defineAbsNode()
1527 graph_node->flags()); in defineAbsNode()
1551 auto graph_node = node->xnode_union_as_XNNPReLU(); in definePReLUNode() local
1555 remapped_ids.at(graph_node->input1_id()), in definePReLUNode()
1556 remapped_ids.at(graph_node->input2_id()), in definePReLUNode()
1557 remapped_ids.at(graph_node->output_id()), in definePReLUNode()
1558 graph_node->flags()); in definePReLUNode()
1582 auto graph_node = node->xnode_union_as_XNNConcatenate2(); in defineConcatenate2Node() local
1586 graph_node->axis(), in defineConcatenate2Node()
1587 remapped_ids.at(graph_node->input1_id()), in defineConcatenate2Node()
1588 remapped_ids.at(graph_node->input2_id()), in defineConcatenate2Node()
1589 remapped_ids.at(graph_node->output_id()), in defineConcatenate2Node()
1590 graph_node->flags()); in defineConcatenate2Node()
1614 auto graph_node = node->xnode_union_as_XNNConcatenate3(); in defineConcatenate3Node() local
1618 graph_node->axis(), in defineConcatenate3Node()
1619 remapped_ids.at(graph_node->input1_id()), in defineConcatenate3Node()
1620 remapped_ids.at(graph_node->input2_id()), in defineConcatenate3Node()
1621 remapped_ids.at(graph_node->input3_id()), in defineConcatenate3Node()
1622 remapped_ids.at(graph_node->output_id()), in defineConcatenate3Node()
1623 graph_node->flags()); in defineConcatenate3Node()
1647 auto graph_node = node->xnode_union_as_XNNConcatenate4(); in defineConcatenate4Node() local
1651 graph_node->axis(), in defineConcatenate4Node()
1652 remapped_ids.at(graph_node->input1_id()), in defineConcatenate4Node()
1653 remapped_ids.at(graph_node->input2_id()), in defineConcatenate4Node()
1654 remapped_ids.at(graph_node->input3_id()), in defineConcatenate4Node()
1655 remapped_ids.at(graph_node->input4_id()), in defineConcatenate4Node()
1656 remapped_ids.at(graph_node->output_id()), in defineConcatenate4Node()
1657 graph_node->flags()); in defineConcatenate4Node()
1681 auto graph_node = node->xnode_union_as_XNNStaticSlice(); in defineStaticSliceNode() local
1683 std::vector<size_t> offsets = flatbufferDimsToVector(graph_node->offsets()); in defineStaticSliceNode()
1684 std::vector<size_t> sizes = flatbufferDimsToVector(graph_node->sizes()); in defineStaticSliceNode()
1688 graph_node->num_dims(), in defineStaticSliceNode()
1691 remapped_ids.at(graph_node->input_id()), in defineStaticSliceNode()
1692 remapped_ids.at(graph_node->output_id()), in defineStaticSliceNode()
1693 graph_node->flags()); in defineStaticSliceNode()
1717 auto graph_node = node->xnode_union_as_XNNScaledDotProductAttention(); in defineScaledDotProductAttentionNode() local
1723 remapped_ids.at(graph_node->query_id()), in defineScaledDotProductAttentionNode()
1724 remapped_ids.at(graph_node->key_id()), in defineScaledDotProductAttentionNode()
1725 remapped_ids.at(graph_node->value_id()), in defineScaledDotProductAttentionNode()
1726 remapped_ids.at(graph_node->scale_id()), in defineScaledDotProductAttentionNode()
1727 remapped_ids.at(graph_node->mask_id()), in defineScaledDotProductAttentionNode()
1728 remapped_ids.at(graph_node->output_id()), in defineScaledDotProductAttentionNode()
1729 graph_node->flags()); in defineScaledDotProductAttentionNode()
1753 auto graph_node = node->xnode_union_as_XNNBatchMatrixMultiply(); in defineBatchMatrixMultiplyNode() local
1757 remapped_ids.at(graph_node->input1_id()), in defineBatchMatrixMultiplyNode()
1758 remapped_ids.at(graph_node->input2_id()), in defineBatchMatrixMultiplyNode()
1759 remapped_ids.at(graph_node->output_id()), in defineBatchMatrixMultiplyNode()
1760 graph_node->flags()); in defineBatchMatrixMultiplyNode()