Searched refs:StackLayerNode (Results 1 – 8 of 8) sorted by relevance
38 StackLayerNode::StackLayerNode(unsigned int total_nodes, int axis) in StackLayerNode() function in arm_compute::graph::StackLayerNode45 int StackLayerNode::axis() const in axis()50 TensorDescriptor StackLayerNode::compute_output_descriptor(const std::vector<TensorDescriptor> &inp… in compute_output_descriptor()65 bool StackLayerNode::forward_descriptors() in forward_descriptors()77 TensorDescriptor StackLayerNode::configure_output(size_t idx) const in configure_output()105 NodeType StackLayerNode::type() const in type()110 void StackLayerNode::accept(INodeVisitor &v) in accept()
34 class StackLayerNode final : public INode42 StackLayerNode(unsigned int total_nodes, int axis);
74 class StackLayerNode; variable
178 virtual void visit(StackLayerNode &n) = 0;302 virtual void visit(StackLayerNode &) override in visit() argument
198 …ail::create_stack_layer<NEStackLayer, NETargetInfo>(*polymorphic_downcast<StackLayerNode *>(node)); in create()
315 …ail::create_stack_layer<CLStackLayer, CLTargetInfo>(*polymorphic_downcast<StackLayerNode *>(node)); in create()
732 …return create_simple_multiple_input_single_output_node<StackLayerNode>(g, params, inputs, inputs.s… in add_stack_node()
1769 std::unique_ptr<arm_compute::IFunction> create_stack_layer(StackLayerNode &node) in create_stack_layer()