Poplar and PopLibs
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Broadcast.hpp
Go to the documentation of this file.
1// Copyright (c) 2018 Graphcore Ltd. All rights reserved.
8#ifndef poputil_Broadcast_hpp
9#define poputil_Broadcast_hpp
10#include <poplar/Tensor.hpp>
11
12namespace poputil {
13
23void expandToMatchRanks(poplar::Tensor &a, poplar::Tensor &b);
24
43void broadcastToMatch(poplar::Tensor &a, const std::vector<std::size_t> &shape);
44
65void broadcastToMatch(poplar::Tensor &a, poplar::Tensor &b);
66
89void broadcastToMatch(poplar::Tensor &a, poplar::Tensor &b, poplar::Tensor &c);
90
101bool canBroadcastToMatch(const poplar::Tensor &a, const poplar::Tensor &b);
102
103} // namespace poputil
104
105#endif // poputil_Broadcast_hpp
poplar::Tensor
A reference to a subset of tensor elements.
Definition: Tensor.hpp:38
poputil
General utility functions for building graphs.
Definition: GfloatExprUtil.hpp:23
poputil::canBroadcastToMatch
bool canBroadcastToMatch(const poplar::Tensor &a, const poplar::Tensor &b)
Test if the given tensors can be broadcast to match one another using the rules for broadcastToMatch(...
poputil::expandToMatchRanks
void expandToMatchRanks(poplar::Tensor &a, poplar::Tensor &b)
Match dimensions of two tensors using numpy-style expansion rules.
poputil::broadcastToMatch
void broadcastToMatch(poplar::Tensor &a, const std::vector< std::size_t > &shape)
Match dimensions of a tensor to a shape using numpy-style broadcast rules: