File tensor_operators.h

↰ Parent directory (src/tensors)

Definition (src/tensors/tensor_operators.h)

• Program Listing for File tensor_operators.h

Includes

• algorithm (File algorithm.cu)

• common/definitions.h (File definitions.h)

• functional/shape.h (File shape.h)

• functional/tensor.h (File tensor.h)

• functional/tmp.h (File tmp.h)

• tensors/allocator.h (File allocator.h)

• tensors/cpu/add.h (File add.h)

• tensors/cpu/element.h (File element.h)

• tensors/dispatch.h (File dispatch.h)

• tensors/tensor.h (File tensor.h)

Included By

• File expression_graph.cpp

• File node_initializers.cpp

• File node_initializers.h

• File node_operators.cpp

• File node_operators_binary.h

• File node_operators_unary.h

• File lsh.cpp

• File clippers.cpp

• File exponential_smoothing.cpp

• File exponential_smoothing.h

• File optimizers.cpp

• File quantizer.cpp

• File quantizer.h

• File attention.cpp

• File cells.cpp

• File packed_gemm.cpp

• File integer_common.h

• File tensor_operators.cpp

• File topk.cpp

• File add_all.cu

• File add_all.h

• File algorithm.cu

• File prod.h

• File tensor_operators.cu

• File topk.cu

• File rand.cpp

• File tensor.cpp

• File communicator.h

• File communicator_nccl.h

• File graph_group_async.cpp

Namespaces

• Namespace marian

• Namespace marian::cpu

Functions

• Template Function marian::Add(Functor, float, marian::Tensor, Tensors…)

• Template Function marian::Add(Functor, marian::Tensor, Tensors…)

• Template Function marian::Aggregate

• Template Function marian::copy

• Function marian::cpu::Deconcatenate(std::vector<marian::Tensor>&, const marian::Tensor, int)

• Function marian::cpu::LSTMCellBackward(std::vector<marian::Tensor>, std::vector<marian::Tensor>, marian::Tensor)

• Function marian::cpu::LSTMOutputBackward(std::vector<marian::Tensor>, std::vector<marian::Tensor>, marian::Tensor)

• Function marian::Deconcatenate

• Function marian::DISPATCH2(Softmax, marian::Tensor, marian::Tensor)

• Function marian::DISPATCH2(LSTMCellForward, marian::Tensor, std::vector<marian::Tensor>)

• Function marian::DISPATCH2(LogSoftmax, marian::Tensor, marian::Tensor)

• Function marian::DISPATCH2(CopyCast, marian::Tensor, const marian::Tensor)

• Function marian::DISPATCH2(AddCast, marian::Tensor, const marian::Tensor)

• Function marian::DISPATCH3(GRUFastForward, marian::Tensor, std::vector<marian::Tensor>, bool)

• Function marian::DISPATCH3(TransposeND, marian::Tensor, marian::Tensor, const std::vector<int>&)

• Function marian::DISPATCH3(Concatenate, marian::Tensor, const std::vector<marian::Tensor>&, int)

• Function marian::DISPATCH3(CopyRows, marian::Tensor, const marian::Tensor, const marian::Tensor)

• Function marian::DISPATCH3(CopyCols, marian::Tensor, const marian::Tensor, const marian::Tensor)

• Function marian::DISPATCH4(CrossEntropyPick, marian::Tensor, marian::Tensor, marian::Tensor, float)

• Function marian::DISPATCH4(IsNaN, const, Ptr<Allocator>, bool&, bool&)

• Function marian::DISPATCH4(Att, marian::Tensor, marian::Tensor, marian::Tensor, marian::Tensor)

• Function marian::DISPATCH4(HighwayForward, marian::Tensor, const marian::Tensor, const marian::Tensor, const marian::Tensor)

• Function marian::DISPATCH4(Select, marian::Tensor, const marian::Tensor, const marian::Tensor, int)

• Function marian::DISPATCH5(Shift, marian::Tensor, marian::Tensor, marian::Shape, float, bool)

• Function marian::DISPATCH5(PoolingWithMaskingForward, marian::Tensor, marian::Tensor, marian::Tensor, int, bool)

• Function marian::DISPATCH5(LayerNormalization, marian::Tensor, marian::Tensor, marian::Tensor, marian::Tensor, float)

• Function marian::DISPATCH5(RMSNormalization, marian::Tensor, marian::Tensor, marian::Tensor, marian::Tensor, float)

• Function marian::DISPATCH7(TopK, marian::Tensor, marian::Tensor, Ptr<Allocator>, const marian::Tensor, int, int, bool)

• Function marian::DISPATCH7(Prod, marian::Tensor, const marian::Tensor&, const marian::Tensor&, bool, bool, float, float)

• Function marian::DISPATCH8

• Function marian::DISPATCH9

• Template Function marian::Element

• Function marian::GRUFastBackward

• Template Function marian::Insert

• Function marian::L2Norm

• Function marian::LayerNormalizationGrad

• Function marian::LSTMCellBackward

• Function marian::LSTMOutputBackward

• Template Function marian::Reduce(Functor, float, marian::Tensor, Tensors…)

• Template Function marian::Reduce(Functor, marian::Tensor, Tensors…)

• Template Function marian::Reduce(Functor, AggFunctor, float, marian::Tensor, Tensors…)

• Function marian::RMSNormalizationGrad

• Function marian::SanitizeGradient

Variables

• Variable marian::bool

• Variable marian::float

• Variable marian::int

• Variable marian::keepProb

• Template Variable marian::Ptr< Allocator >

• Variable marian::scale

• Variable marian::shift