| 83 | } |
| 84 | |
| 85 | std::vector<ComputeTaskDescriptorPtr> SelectDepthWiseConv( |
| 86 | int id, ValueId input_id, ValueId output_id, |
| 87 | const DepthwiseConvolution2DAttributes& attr, |
| 88 | const metal::RuntimeOptions& options) { |
| 89 | if (CheckDepthWiseConv3x3Stride1x1Support(attr)) { |
| 90 | return DepthWiseConv3x3Stride1x1(id, input_id, output_id, attr, options); |
| 91 | } else if (CheckDepthWiseConv3x3Stride2Support(attr)) { |
| 92 | return DepthWiseConv3x3Stride2(id, input_id, output_id, attr, options); |
| 93 | } else { |
| 94 | return DepthWiseConvolution(id, input_id, output_id, attr, options); |
| 95 | } |
| 96 | } |
| 97 | |
| 98 | std::vector<ComputeTaskDescriptorPtr> SelectReshape( |
| 99 | const GraphFloat32& graph, int id, ValueId input_id, ValueId output_id, |
no test coverage detected