Lines Matching refs:xnnpackTensors
214 std::vector<uint32_t> xnnpackTensors(operands.size()); in Create() local
246 static_cast<uint32_t>(t), flags, &xnnpackTensors[t]); in Create()
255 if (VisitNode(subgraph.get(), operation, operands.data(), xnnpackTensors) != in Create()
498 const std::vector<uint32_t>& xnnpackTensors) { in VisitNode() argument
501 return VisitAbsNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
503 return VisitAddNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
505 return VisitAveragePool2DNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
507 return VisitConv2DNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
509 return VisitDepthwiseConv2DNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
511 return VisitDivNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
513 return VisitFloorNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
515 return VisitFullyConnectedNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
517 return VisitHardSwishNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
519 return VisitLogisticNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
521 return VisitMaxPool2DNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
523 return VisitMaximumNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
525 return VisitMeanNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
527 return VisitMinimumNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
529 return VisitMulNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
531 return VisitNegNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
533 return VisitPadNode(subgraph, operation, operands, 0.0f, xnnpackTensors); in VisitNode()
535 return VisitPadV2Node(subgraph, operation, operands, xnnpackTensors); in VisitNode()
537 return VisitReshapeNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
539 return VisitResizeBilinearNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
541 return VisitPreluNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
544 std::numeric_limits<float>::infinity(), xnnpackTensors); in VisitNode()
546 return VisitReluNode(subgraph, operation, operands, -1.0f, 1.0f, xnnpackTensors); in VisitNode()
548 return VisitReluNode(subgraph, operation, operands, 0.0f, 6.0f, xnnpackTensors); in VisitNode()
550 return VisitSqrtNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
552 return VisitSubNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
554 return VisitSoftmaxNode(subgraph, operation, operands, xnnpackTensors); in VisitNode()
562 const std::vector<uint32_t>& xnnpackTensors) { in VisitAbsNode() argument
570 xnn_define_abs(subgraph, /*input_id=*/xnnpackTensors[ins[0]], in VisitAbsNode()
571 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitAbsNode()
582 const std::vector<uint32_t>& xnnpackTensors) { in VisitAddNode() argument
599 /*input1_id=*/xnnpackTensors[ins[0]], in VisitAddNode()
600 /*input2_id=*/xnnpackTensors[ins[1]], in VisitAddNode()
601 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitAddNode()
613 const std::vector<uint32_t>& xnnpackTensors) { in VisitAveragePool2DNode() argument
674 /*input_id=*/xnnpackTensors[ins[0]], in VisitAveragePool2DNode()
675 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitAveragePool2DNode()
682 /*input_id=*/xnnpackTensors[ins[0]], in VisitAveragePool2DNode()
683 /*output_id=*/xnnpackTensors[outs[0]], flags); in VisitAveragePool2DNode()
696 const std::vector<uint32_t>& xnnpackTensors) { in VisitConv2DNode() argument
779 /*input_id=*/xnnpackTensors[ins[0]], in VisitConv2DNode()
780 /*filter_id=*/xnnpackTensors[ins[1]], in VisitConv2DNode()
781 /*bias_id=*/xnnpackTensors[ins[2]], in VisitConv2DNode()
782 /*output_id=*/xnnpackTensors[outs[0]], flags); in VisitConv2DNode()
795 const std::vector<uint32_t>& xnnpackTensors) { in VisitDepthwiseConv2DNode() argument
881 /*input_id=*/xnnpackTensors[ins[0]], in VisitDepthwiseConv2DNode()
882 /*filter_id=*/xnnpackTensors[ins[1]], in VisitDepthwiseConv2DNode()
883 /*bias_id=*/xnnpackTensors[ins[2]], in VisitDepthwiseConv2DNode()
884 /*output_id=*/xnnpackTensors[outs[0]], flags); in VisitDepthwiseConv2DNode()
895 const std::vector<uint32_t>& xnnpackTensors) { in VisitDivNode() argument
912 /*input1_id=*/xnnpackTensors[ins[0]], in VisitDivNode()
913 /*input2_id=*/xnnpackTensors[ins[1]], in VisitDivNode()
914 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitDivNode()
926 const std::vector<uint32_t>& xnnpackTensors) { in VisitFullyConnectedNode() argument
946 /*input_id=*/xnnpackTensors[ins[0]], in VisitFullyConnectedNode()
947 /*filter_id=*/xnnpackTensors[ins[1]], in VisitFullyConnectedNode()
948 /*bias_id=*/xnnpackTensors[ins[2]], in VisitFullyConnectedNode()
949 /*output_id=*/xnnpackTensors[outs[0]], in VisitFullyConnectedNode()
962 const std::vector<uint32_t>& xnnpackTensors) { in VisitFloorNode() argument
971 /*input1_id=*/xnnpackTensors[ins[0]], in VisitFloorNode()
972 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitFloorNode()
984 const std::vector<uint32_t>& xnnpackTensors) { in VisitHardSwishNode() argument
992 xnn_define_hardswish(subgraph, /*input_id=*/xnnpackTensors[ins[0]], in VisitHardSwishNode()
993 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitHardSwishNode()
1005 const std::vector<uint32_t>& xnnpackTensors) { in VisitLogisticNode() argument
1013 xnn_define_sigmoid(subgraph, /*input_id=*/xnnpackTensors[ins[0]], in VisitLogisticNode()
1014 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitLogisticNode()
1026 const std::vector<uint32_t>& xnnpackTensors) { in VisitMaxPool2DNode() argument
1087 /*input_id=*/xnnpackTensors[ins[0]], in VisitMaxPool2DNode()
1088 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitMaxPool2DNode()
1096 /*input_id=*/xnnpackTensors[ins[0]], in VisitMaxPool2DNode()
1097 /*output_id=*/xnnpackTensors[outs[0]], flags); in VisitMaxPool2DNode()
1110 const std::vector<uint32_t>& xnnpackTensors) { in VisitMaximumNode() argument
1127 /*input1_id=*/xnnpackTensors[ins[0]], in VisitMaximumNode()
1128 /*input2_id=*/xnnpackTensors[ins[1]], in VisitMaximumNode()
1129 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitMaximumNode()
1141 const std::vector<uint32_t>& xnnpackTensors) { in VisitMeanNode() argument
1172 /*input_id=*/xnnpackTensors[ins[0]], in VisitMeanNode()
1173 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitMeanNode()
1185 const std::vector<uint32_t>& xnnpackTensors) { in VisitMinimumNode() argument
1202 /*input1_id=*/xnnpackTensors[ins[0]], in VisitMinimumNode()
1203 /*input2_id=*/xnnpackTensors[ins[1]], in VisitMinimumNode()
1204 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitMinimumNode()
1215 const std::vector<uint32_t>& xnnpackTensors) { in VisitMulNode() argument
1232 /*input1_id=*/xnnpackTensors[ins[0]], in VisitMulNode()
1233 /*input2_id=*/xnnpackTensors[ins[1]], in VisitMulNode()
1234 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitMulNode()
1245 const std::vector<uint32_t>& xnnpackTensors) { in VisitNegNode() argument
1254 /*input1_id=*/xnnpackTensors[ins[0]], in VisitNegNode()
1255 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitNegNode()
1267 const std::vector<uint32_t>& xnnpackTensors) { in VisitPreluNode() argument
1281 xnn_define_prelu(subgraph, /*input_id=*/xnnpackTensors[ins[0]], in VisitPreluNode()
1282 /*slope_id=*/xnnpackTensors[ins[1]], in VisitPreluNode()
1283 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitPreluNode()
1294 const std::vector<uint32_t>& xnnpackTensors) { in VisitPadNode() argument
1318 /*input_id=*/xnnpackTensors[ins[0]], in VisitPadNode()
1319 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitPadNode()
1331 const std::vector<uint32_t>& xnnpackTensors) { in VisitPadV2Node() argument
1337 return VisitPadNode(subgraph, operation, operands, padding_value, xnnpackTensors); in VisitPadV2Node()
1343 const std::vector<uint32_t>& xnnpackTensors) { in VisitReshapeNode() argument
1362 /*input_id=*/xnnpackTensors[ins[0]], in VisitReshapeNode()
1363 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitReshapeNode()
1375 const std::vector<uint32_t>& xnnpackTensors) { in VisitResizeBilinearNode() argument
1432 /*input_id=*/xnnpackTensors[ins[0]], in VisitResizeBilinearNode()
1433 /*output_id=*/xnnpackTensors[outs[0]], flags); in VisitResizeBilinearNode()
1446 const std::vector<uint32_t>& xnnpackTensors) { in VisitReluNode() argument
1455 /*input_id=*/xnnpackTensors[ins[0]], in VisitReluNode()
1456 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitReluNode()
1468 const std::vector<uint32_t>& xnnpackTensors) { in VisitSqrtNode() argument
1477 /*input1_id=*/xnnpackTensors[ins[0]], in VisitSqrtNode()
1478 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitSqrtNode()
1489 const std::vector<uint32_t>& xnnpackTensors) { in VisitSubNode() argument
1506 /*input1_id=*/xnnpackTensors[ins[0]], in VisitSubNode()
1507 /*input2_id=*/xnnpackTensors[ins[1]], in VisitSubNode()
1508 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitSubNode()
1520 const std::vector<uint32_t>& xnnpackTensors) { in VisitSoftmaxNode() argument
1542 xnn_define_softmax(subgraph, /*input_id=*/xnnpackTensors[ins[0]], in VisitSoftmaxNode()
1543 /*output_id=*/xnnpackTensors[outs[0]], /*flags=*/0); in VisitSoftmaxNode()