diff options
Diffstat (limited to 'runtimes/neurun/backend/cpu/KernelGenerator.cc')
-rw-r--r-- | runtimes/neurun/backend/cpu/KernelGenerator.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/runtimes/neurun/backend/cpu/KernelGenerator.cc b/runtimes/neurun/backend/cpu/KernelGenerator.cc index 5def6c940..34868ba3a 100644 --- a/runtimes/neurun/backend/cpu/KernelGenerator.cc +++ b/runtimes/neurun/backend/cpu/KernelGenerator.cc @@ -324,6 +324,8 @@ void KernelGenerator::visit(const model::operation::PermuteNode &node) const auto output_backend = node.param().output_backend; const auto data_type = node.getDataType(); + output_backend->tensor_builder()->preVisit(node); + auto output_object = output_backend->tensor_builder()->wrapTensor(output_index); auto input_object = input_backend->tensor_builder()->wrapTensor(input_index); @@ -358,6 +360,8 @@ void KernelGenerator::visit(const model::operation::PermuteNode &node) fn->configure(input_object, output_object, out_shape, permuteType, data_type); + input_backend->tensor_builder()->postVisit(node); + _execution_builder->append(std::move(fn)); } |