MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / get_elem_size

Method get_elem_size

lite/src/tensor.cpp:15–39  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

13using namespace lite;
14
15size_t Layout::get_elem_size() const {
16 size_t elesize = 1;
17 switch (data_type) {
18 case LiteDataType::LITE_INT64:
19 elesize = 8;
20 break;
21 case LiteDataType::LITE_FLOAT:
22 case LiteDataType::LITE_INT:
23 case LiteDataType::LITE_UINT:
24 elesize = 4;
25 break;
26 case LiteDataType::LITE_HALF:
27 case LiteDataType::LITE_INT16:
28 case LiteDataType::LITE_UINT16:
29 elesize = 2;
30 break;
31 case LiteDataType::LITE_INT8:
32 case LiteDataType::LITE_UINT8:
33 elesize = 1;
34 break;
35 default:
36 LITE_THROW("not support data type.");
37 }
38 return elesize;
39}
40
41bool Layout::operator==(const Layout& other) const {
42 bool equal = true;

Callers 15

device_inputFunction · 0.80
device_input_outputFunction · 0.80
pinned_host_inputFunction · 0.80
config_user_allocatorFunction · 0.80
output_data_infoFunction · 0.80
basic_load_from_memoryFunction · 0.80
async_forwardFunction · 0.80
set_input_callbackFunction · 0.80
set_output_callbackFunction · 0.80
reset_inputFunction · 0.80
register_cryption_methodFunction · 0.80

Calls

no outgoing calls

Tested by 3

test_io_no_copy_axFunction · 0.64
TESTFunction · 0.64
compare_lite_tensorFunction · 0.64