Poplar and PopLibs
ConvPreplan.hpp
Go to the documentation of this file.
1// Copyright (c) 2021 Graphcore Ltd. All rights reserved.
8#ifndef poplin_ConvPreplan_hpp
9#define poplin_ConvPreplan_hpp
10#include "Convolution.hpp"
11#include "MatMul.hpp"
12
13#include <poplar/Graph.hpp>
14#include <poplar/OptionFlags.hpp>
15#include <set>
16#include <tuple>
17
18namespace poplin {
19
37void preplan(const std::set<ConvPlanParams> &convs,
38 const std::set<MatMulPlanParams> &matmuls, PlanningCache &cache);
39
40} // namespace poplin
41
42#endif // poplin_ConvPreplan_hpp
Convolution.hpp
Functions and data types to support performing convolutions.
poplin
Linear algebra functions.
Definition: Cholesky.hpp:14
poplin::preplan
void preplan(const std::set< ConvPlanParams > &convs, const std::set< MatMulPlanParams > &matmuls, PlanningCache &cache)
Plan the specified convolutions & matmuls.
MatMul.hpp
Sparse matrix multiply operations.