9 #include "ade/ifunctor.hpp" 11 #ifndef LLO_HELPER_HPP 12 #define LLO_HELPER_HPP 18 ade::TensptrT
mtens_mul (ade::TensptrT lhs, ade::MappedTensor rhs);
33 ade::TensptrT
reduce (ade::Opcode opcode, ade::TensptrT tens, uint8_t dim);
36 ade::TensptrT
matmul (ade::TensptrT a, ade::TensptrT b);
39 ade::TensptrT
convolution (ade::TensptrT img, ade::TensptrT kernel);
43 #endif // LLO_HELPER_HPP ade::TensptrT mtens_mul(ade::TensptrT lhs, ade::MappedTensor rhs)
Return product of lhs with mapped rhs.
Definition: helper.cpp:11
ade::TensptrT grad_prod(ade::iFunctor *fwd, size_t gradidx, ade::TensT tens)
Definition: helper.cpp:18
ade::TensptrT convolution(ade::TensptrT img, ade::TensptrT kernel)
Return convolution operation on img with kernel.
Definition: helper.cpp:102
ade::TensptrT matmul(ade::TensptrT a, ade::TensptrT b)
Return matmul of a and b.
Definition: helper.cpp:67
std::vector< ade::TensptrT > TensT
Tensptr vector type.
Definition: data.hpp:22
ade::TensptrT grad_max(ade::iFunctor *fwd, size_t gradidx, ade::TensT tens)
Definition: helper.cpp:46
ade::TensptrT grad_min(ade::iFunctor *fwd, size_t gradidx, ade::TensT tens)
Definition: helper.cpp:35
ade::TensptrT reduce(ade::Opcode opcode, ade::TensptrT tens, uint8_t dim)
Return reduction of tens after dimension dim using opcode operation.
Definition: helper.cpp:57