Lines Matching refs:CheckTensorShape
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()
1352 CheckTensorShape(operands[outs[0]].dimensions, 0, XNN_MAX_TENSOR_DIMS)); in VisitReshapeNode()
1379 NN_DRIVER_RETURN_IF_ERROR(CheckTensorShape(operands[ins[0]].dimensions, 4)); in VisitResizeBilinearNode()
1381 NN_DRIVER_RETURN_IF_ERROR(CheckTensorShape(operands[outs[0]].dimensions, 4)); in VisitResizeBilinearNode()