Poplar and PopLibs
ElementWiseUtil.hpp
Go to the documentation of this file.
1// Copyright (c) 2019 Graphcore Ltd. All rights reserved.
8#ifndef _popops_ElementWiseUtil_hpp_
9#define _popops_ElementWiseUtil_hpp_
10
11#include <poplar/Graph.hpp>
12
13#include <vector>
14
15namespace popops {
16
37poplar::Tensor
38createOutputForElementWiseOp(poplar::Graph &graph,
39 const std::vector<poplar::Tensor> &inputs,
40 const poplar::Type &outputType,
41 const poplar::DebugContext &debugContext = {});
42
43} // end namespace popops
44
45#endif // _popops_ElementWiseUtil_hpp_
poplar::DebugContext
DebugContext gathers the common external parameters of the context of an operation.
Definition: DebugContext.hpp:221
poplar::Graph
This class represents a graph program to be executed on the IPU.
Definition: Graph.hpp:52
poplar::Tensor
A reference to a subset of tensor elements.
Definition: Tensor.hpp:38
poplar::Type
Class representing device data types.
Definition: Type.hpp:42
popops
Common functions, such as elementwise and reductions.
Definition: AllTrue.hpp:15
popops::createOutputForElementWiseOp
poplar::Tensor createOutputForElementWiseOp(poplar::Graph &graph, const std::vector< poplar::Tensor > &inputs, const poplar::Type &outputType, const poplar::DebugContext &debugContext={})
Create a tensor for use as the output of an element-wise operation (operation with no dependency betw...