| 96 | } |
| 97 | |
| 98 | std::vector<ComputeTaskDescriptorPtr> SelectReshape( |
| 99 | const GraphFloat32& graph, int id, ValueId input_id, ValueId output_id, |
| 100 | const ReshapeAttributes& attr) { |
| 101 | const auto src_shape = graph.FindInputs(id)[0]->tensor.shape; |
| 102 | if (src_shape.c % 4 == 0 && attr.new_shape.c % 4 == 0) { |
| 103 | return Reshapex4(id, input_id, output_id, attr); |
| 104 | } else { |
| 105 | return Reshape(id, input_id, output_id, attr); |
| 106 | } |
| 107 | } |
| 108 | |
| 109 | std::vector<ComputeTaskDescriptorPtr> SelectSoftmax(const GraphFloat32& graph, |
| 110 | int id, ValueId input_id, |
no test coverage detected