Poplar and PopLibs
FullyConnected.hpp
Go to the documentation of this file.
1// Copyright (c) 2019 Graphcore Ltd. All rights reserved.
9#ifndef poplin_FullyConnected_hpp
10#define poplin_FullyConnected_hpp
11
12#include "poplin/MatMul.hpp"
13
14namespace poplin {
15namespace fc {
16
18struct FullyConnectedParams {
23 std::size_t numGroups;
25 std::size_t batchSize;
27 std::size_t inputSize;
29 std::size_t outputSize;
30};
31
49std::vector<std::pair<MatMulParams, poplar::OptionFlags>>
50getMatMulPrePlanParameters(FullyConnectedParams parameters,
51 poplar::OptionFlags matmulOptions, poplar::Type type,
52 bool inferenceOnly);
53
54} // namespace fc
55} // namespace poplin
56
57#endif // poplin_FullyConnected_hpp
poplar::OptionFlags
A set of option/value string flags to be used in various APIs.
Definition: OptionFlags.hpp:24
poplar::Type
Class representing device data types.
Definition: Type.hpp:42
poplin
Linear algebra functions.
Definition: Cholesky.hpp:14
poplin::fc::getMatMulPrePlanParameters
std::vector< std::pair< MatMulParams, poplar::OptionFlags > > getMatMulPrePlanParameters(FullyConnectedParams parameters, poplar::OptionFlags matmulOptions, poplar::Type type, bool inferenceOnly)
Predict what matrix multiplications will be needed for the given parameters and return a list of corr...
MatMul.hpp
Functions and data types for performing matrix multiplies on the IPU.
poplin::fc::FullyConnectedParams
Parameters to describe a fully-connected layer.
Definition: FullyConnected.hpp:18
poplin::fc::FullyConnectedParams::outputSize
std::size_t outputSize
Size of the output in each batch from the layer.
Definition: FullyConnected.hpp:29
poplin::fc::FullyConnectedParams::numGroups
std::size_t numGroups
The number of groups (where each group represents a fully-connected layer of the same shape) that ope...
Definition: FullyConnected.hpp:23
poplin::fc::FullyConnectedParams::inputSize
std::size_t inputSize
Size of the input in each batch into the layer.
Definition: FullyConnected.hpp:27
poplin::fc::FullyConnectedParams::batchSize
std::size_t batchSize
Number of samples in the input to the layer.
Definition: FullyConnected.hpp:25