29 lines
539 B
C++
29 lines
539 B
C++
|
#pragma once
|
||
|
|
|
||
|
|
#include "tensor.hpp"
|
||
|
|
|
||
|
|
namespace tf {
|
||
|
|
|
||
|
|
// TODO
|
||
|
|
template <typename T>
|
||
|
|
void tensor_add (Tensor<T>& res, Tensor<T>& lhs, Tensor<T>& rhs) {
|
||
|
|
|
||
|
|
if(res._shape != lhs._shape || lhs._shape != rhs._shape) {
|
||
|
|
TF_THROW("tensor shapes do not match!");
|
||
|
|
}
|
||
|
|
|
||
|
|
// case 1: all tensors have data in memory
|
||
|
|
if(res._storage_level == MEMORY &&
|
||
|
|
lhs._storage_level == MEMORY &&
|
||
|
|
rhs._storage_level == MEMORY) {
|
||
|
|
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
// case 2: TODO
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
} // end of namespace tf -----------------------------------------------------
|