This PR includes changes of Tensor and TensorDim to support NHWC
computation for dot, add_strided, multiply_strided, cat, split,
and transpose. It also includes unittests to evaluate.
**Self evaluation:**
1. Build test: [X]Passed [ ]Failed [ ]Skipped
2. Run test: [X]Passed [ ]Failed [ ]Skipped
Signed-off-by: Adwaith Anand <adwaith.a@samsung.com>
Signed-off-by: Manohara HK <manohara.hk@samsung.com>
Signed-off-by: jijoong.moon <jijoong.moon@samsung.com>
Tensor::Tensor(const TensorDim &d, bool alloc_now, Tensor::Initializer init,
std::string name_) :
- Tensor(name_) {
+ Tensor(name_, d.getFormat()) {
if (d.getDataLen() != 0) {
dim = d;
strides = d.computeStrides();
/**
* @brief Constructor of Tensor
* @note This constructor copies vector again. needs refactoring
- * @param[in] d data for the Tensor
+ * @param[in] d data for the Tensor. It needs to set format properly.
*/
Tensor(std::vector<std::vector<std::vector<_FP16>>> const &d,
ml::train::TensorDim::TensorType t_type) :