Struct Splitter

• Defined in File splitter.hpp

Struct Documentation

struct ripple::Splitter

This class splits a tensor by its partitions, creating a node in the graph for each of the partitions in the tensor.

It also determines the dependencies between nodes in the graph, and adds memory transfer nodes for copying tensor padding data, if required.

Public Static Functions

template<typename Graph, typename F, typename ...Args>
auto split(Graph &graph, ExecutionKind exe, F &&functor, Args&&... args) noexcept -> void

Creates a split for any argument which is a tensor by creating a node per partition in the tensor.

If any of the arguments have modifiers then the dependencies assosciated with the modifiers are added when adding the nodes, as well as any additional nodes (such as memcpy nodes for padding).

Parameters

• graph: The graph to add the split nodes to.

• functor: The functor which defines the operations on the nodes.

• args: The arguments for the functor.

Template Parameters

• Graph: The type of the graph.

• F: The type of the functor.

• Args: The type of the arguments.