47 #ifdef ARM_COMPUTE_ASSERTS_ENABLED 50 auto print_node = utils::cast::polymorphic_downcast<PrintLayerNode *>(task.
node);
51 auto input_handle = print_node->input(0)->handle();
52 auto transform = print_node->transform();
54 input_handle->map(
true);
55 ITensor *
input = transform ? transform(&input_handle->tensor()) : &input_handle->tensor();
56 input->
print(print_node->stream(), print_node->format_info());
57 input_handle->unmap();
59 #endif // ARM_COMPUTE_ASSERTS_ENABLED 70 TaskExecutor::TaskExecutor()
void operator()()
Function operator.
void prepare()
Prepare execution task.
Interface for CPU tensor.
Copyright (c) 2017-2022 Arm Limited.
std::unique_ptr< arm_compute::IFunction > task
Task to execute.
void execute_task(ExecutionTask &task)
static TaskExecutor & get()
Task executor accessor.
void print(std::ostream &s, IOFormatInfo io_fmt=IOFormatInfo()) const
Print a tensor to a given stream using user defined formatting information.
virtual NodeType type() const =0
Returns node's type.
INode * node
Node bound to this workload.
std::function< decltype(execute_task)> execute_function
Function that is responsible for executing tasks.