latest/doxygen/Cast_8hpp_source.html

// Copyright (c) 2016 Graphcore Ltd. All rights reserved.

#ifndef popops_Cast_hpp

#define popops_Cast_hpp


#include <poplar/Graph.hpp>

#include <poplar/Interval.hpp>

#include <poplar/Program.hpp>

#include <vector>


namespace popops {


poplar::Tensor cast(poplar::Graph &graph, const poplar::Tensor &src,

                    const poplar::Type &dstType,

                    poplar::program::Sequence &prog,

                    const poplar::DebugContext &debugContext = {});


poplar::Tensor cast(poplar::Graph &graph, const poplar::Tensor &src,

                    const poplar::Type &dstType, const poplar::Tensor &metadata,

                    poplar::program::Sequence &prog,

                    const poplar::DebugContext &debugContext = {});


poplar::program::Program cast(poplar::Graph &graph, poplar::Tensor src,

                              poplar::Tensor dst,

                              const poplar::DebugContext &debugContext = {});


void cast(poplar::Graph &graph, poplar::Tensor src, poplar::Tensor dst,

          poplar::ComputeSet cs);


poplar::Tensor cast(poplar::Graph &graph, poplar::Tensor src,

                    const poplar::Type &dstType, poplar::ComputeSet cs,

                    const poplar::DebugContext &debugContext = {});


poplar::Tensor cast(poplar::Graph &graph, poplar::Tensor src,

                    const poplar::Type &dstType, const poplar::Tensor &metadata,

                    poplar::ComputeSet cs,

                    const poplar::DebugContext &debugContext = {});


poplar::Tensor

checkAccuracyWhenCast(poplar::Graph &graph, const poplar::Tensor &input,

                      poplar::Type outputType, double tolerance,

                      poplar::program::Sequence &prog,

                      const poplar::DebugContext &debugContext = {});


} // namespace popops


#endif // popops_Cast_hpp

poplar::ComputeSet
A reference to a compute set within a graph.
Definition: GraphElements.hpp:131

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

poplar::program::Program
This class represents a control program that executes operations on the graph.
Definition: Program.hpp:30

poplar::program::Sequence
Program that executes a sequence of programs.
Definition: Program.hpp:77

popops
Common functions, such as elementwise and reductions.
Definition: AllTrue.hpp:15

popops::checkAccuracyWhenCast
poplar::Tensor checkAccuracyWhenCast(poplar::Graph &graph, const poplar::Tensor &input, poplar::Type outputType, double tolerance, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
Helper function which checks the relative error in the tensor input when casting it to type outputTyp...

popops::cast
poplar::Tensor cast(poplar::Graph &graph, const poplar::Tensor &src, const poplar::Type &dstType, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
Cast elements of the specified src tensor to dstType, returning the result as a new tensor.