Searched refs:xnnpackTensors (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/driver/sample_hidl/ |
| D | SampleDriverFloatXNNPACK.cpp | 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()
[all …]
|