Poplar and PopLibs
NaN.hpp
Go to the documentation of this file.
1// Copyright (c) 2019 Graphcore Ltd. All rights reserved.
8#ifndef popops_NaN_hpp
9#define popops_NaN_hpp
10
11#include <poplar/Graph.hpp>
12#include <poplar/Program.hpp>
13#include <poplar/Tensor.hpp>
14
15namespace popops {
26poplar::Tensor hasNaN(poplar::Graph &graph, const poplar::Tensor &src,
27 poplar::program::Sequence &prog,
28 const poplar::DebugContext &debugContext = {});
29
40poplar::Tensor hasNaNOrInf(poplar::Graph &graph, const poplar::Tensor &src,
41 poplar::program::Sequence &prog,
42 const poplar::DebugContext &debugContext = {});
43
44} // namespace popops
45
46#endif // popops_NaN_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::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::hasNaN
poplar::Tensor hasNaN(poplar::Graph &graph, const poplar::Tensor &src, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
Test for NaN values in a tensor.
popops::hasNaNOrInf
poplar::Tensor hasNaNOrInf(poplar::Graph &graph, const poplar::Tensor &src, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
Test for NaN or Inf values in a tensor.