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
/*******************************************************************************
* Copyright 2020 Intel Corporation
*
* 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.
*******************************************************************************/
#ifndef GRAPH_INTERFACE_TENSOR_HPP
#define GRAPH_INTERFACE_TENSOR_HPP
#include "graph/interface/c_types_map.hpp"
#include "graph/interface/logical_tensor.hpp"
struct dnnl_graph_tensor {
public:
dnnl_graph_tensor() = default;
dnnl_graph_tensor(const dnnl::impl::graph::logical_tensor_t <,
const dnnl::impl::graph::engine_t *eng, void *handle);
void *get_data_handle() const { return handle_.get(); }
dnnl::impl::graph::status_t set_data_handle(void *handle) {
auto ltw = dnnl::impl::graph::logical_tensor_wrapper_t(lt_);
if (ltw.is_host_scalar()) {
return dnnl::impl::graph::status::invalid_arguments;
} else {
handle_.reset(handle, dummy_destructor);
return dnnl::impl::graph::status::success;
}
}
const dnnl::impl::graph::logical_tensor_t &get_logical_tensor() const {
return lt_;
}
operator bool() const { return handle_ != nullptr; }
const dnnl::impl::graph::engine_t *get_engine() const { return eng_; }
private:
static dnnl::impl::graph::status_t dummy_destructor(void *) {
return dnnl::impl::graph::status::success;
}
dnnl::impl::graph::logical_tensor_t lt_
= dnnl::impl::graph::zero_logical_tensor();
const dnnl::impl::graph::engine_t *eng_ {nullptr};
std::shared_ptr<void> handle_ {nullptr};
union {
// this field is valid when logical tensor's
// property is host_scalar
int32_t s32_value = 0;
float f32_value;
int64_t s64_value;
// TODO: add more dtype support
} scalar_;
};
#endif