1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
/*
* Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "mir/TensorVariant.h"
#include <cstring>
namespace mir
{
TensorVariant::TensorVariant(const TensorType &type) : _type(type), _strides(type.getShape().rank())
{
_element_size = getDataTypeSize(getElementType());
std::size_t data_size = getShape().numElements() * _element_size;
_data.reset(new char[data_size], std::default_delete<char[]>());
int stride = 1;
for (int d = getShape().rank() - 1; d >= 0; --d)
{
_strides[d] = stride;
stride *= getShape().dim(d);
}
}
TensorVariant::TensorVariant(DataType element_type, const Shape &shape)
: TensorVariant(TensorType(element_type, shape))
{
}
TensorVariant::TensorVariant(const TensorType &type, const void *data) : TensorVariant(type)
{
std::size_t data_size = getShape().numElements() * _element_size;
std::memcpy(_data.get(), data, data_size);
}
TensorVariant::TensorVariant(DataType element_type, const Shape &shape, const void *data)
: TensorVariant(TensorType(element_type, shape), data)
{
}
/**
* @brief Construct a TensorVariant from t_old that has strides with 0 where dim = 1
* Used for broadcasting
* @param t_old TensorVariant to use as base
* @param shape shape to broadcast to
*/
TensorVariant::TensorVariant(const TensorVariant &t_old, const Shape &shape)
: _type(t_old.getType().getElementType(), shape), _data(t_old._data),
_strides(static_cast<size_t>(shape.rank())), _element_size(t_old._element_size)
{
int axis_old = t_old.getShape().rank() - 1;
for (int d = shape.rank() - 1; d >= 0; d--)
{
if (axis_old == -1)
break;
if (t_old.getShape().dim(axis_old) != 1)
_strides[d] = t_old._strides[axis_old];
axis_old--;
}
}
} // namespace mir
|
