Searched refs:CheckTensorShape (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/driver/sample_hidl/ |
| D | SampleDriverFloatXNNPACK.cpp | 435 static V1_3::ErrorStatus CheckTensorShape(std::vector<uint32_t>& dimensions, in CheckTensorShape() function in android::nn::sample_driver::Subgraph
454 static V1_3::ErrorStatus CheckTensorShape(std::vector<uint32_t>& dimensions, in CheckTensorShape() function in android::nn::sample_driver::Subgraph
456 return CheckTensorShape(dimensions, expected_num_dims, expected_num_dims); in CheckTensorShape()
1145 NN_DRIVER_RETURN_IF_ERROR(CheckTensorShape(operands[ins[0]].dimensions, 4)); in VisitMeanNode()
1150 NN_DRIVER_RETURN_IF_ERROR(CheckTensorShape(operands[outs[0]].dimensions, 4)); in VisitMeanNode()
1272 CheckTensorShape(operands[ins[0]].dimensions, 1, XNN_MAX_TENSOR_DIMS)); in VisitPreluNode()
1277 CheckTensorShape(operands[outs[0]].dimensions, 1, XNN_MAX_TENSOR_DIMS)); in VisitPreluNode()
1299 CheckTensorShape(operands[ins[0]].dimensions, 1, XNN_MAX_TENSOR_DIMS)); in VisitPadNode()
1303 CheckTensorShape(operands[outs[0]].dimensions, 1, XNN_MAX_TENSOR_DIMS)); in VisitPadNode()
1348 CheckTensorShape(operands[ins[0]].dimensions, 0, XNN_MAX_TENSOR_DIMS)); in VisitReshapeNode()
[all …]
|