latest/doxygen/UpdateScalarInRows_8hpp_source.html

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

#ifndef popops_UpdateScalarInRows_hpp

#define popops_UpdateScalarInRows_hpp

#include <poplar/Graph.hpp>

#include <poplar/Program.hpp>

#include <popops/ExprOp.hpp>


#include <string>


namespace popops {


void updateScalarInRows(poplar::Graph &graph, const poplar::Tensor &params,

                        const poplar::Tensor &indices,

                        poplar::program::Sequence &program,

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


} // namespace popops


#endif // popops_UpdateScalarInRows_hpp

ExprOp.hpp
Operators used in expressions with elements of tensors.

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::updateScalarInRows
void updateScalarInRows(poplar::Graph &graph, const poplar::Tensor &params, const poplar::Tensor &indices, poplar::program::Sequence &program, const poplar::DebugContext &debugContext={})
Update in-place one scalar per row of the tensor params.