|
|
|
@@ -74,10 +74,6 @@ std::vector<lite::tensor::Tensor *> LiteKernelUtil::SubgraphInputTensors( |
|
|
|
for (const auto &kernel : kernels) { |
|
|
|
all_output_tensors.insert(all_output_tensors.end(), kernel->GetOutputs().begin(), kernel->GetOutputs().end()); |
|
|
|
} |
|
|
|
std::sort(all_output_tensors.begin(), all_output_tensors.end()); |
|
|
|
auto end_iter = std::unique(all_output_tensors.begin(), all_output_tensors.end()); |
|
|
|
all_output_tensors.erase(end_iter, all_output_tensors.end()); |
|
|
|
|
|
|
|
std::vector<kernel::LiteKernel *> input_kernels = SubgraphInputKernels(kernels); |
|
|
|
for (const auto &kernel : input_kernels) { |
|
|
|
for (const auto &tensor : kernel->GetInputs()) { |
|
|
|
@@ -97,10 +93,6 @@ std::vector<lite::tensor::Tensor *> LiteKernelUtil::SubgraphOutputTensors( |
|
|
|
for (const auto &kernel : kernels) { |
|
|
|
all_input_tensors.insert(all_input_tensors.end(), kernel->GetInputs().begin(), kernel->GetInputs().end()); |
|
|
|
} |
|
|
|
std::sort(all_input_tensors.begin(), all_input_tensors.end()); |
|
|
|
auto end_iter = std::unique(all_input_tensors.begin(), all_input_tensors.end()); |
|
|
|
all_input_tensors.erase(end_iter, all_input_tensors.end()); |
|
|
|
|
|
|
|
std::vector<kernel::LiteKernel *> output_kernels = SubgraphOutputKernels(kernels); |
|
|
|
for (const auto &kernel : output_kernels) { |
|
|
|
for (const auto &tensor : kernel->GetOutputs()) { |
|
|
|
|