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

Function LITE_get_tensor_layout

lite/lite-c/src/tensor.cpp:193–199  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

191}
192
193int LITE_get_tensor_layout(const LiteTensor tensor, LiteLayout* layout) {
194 LITE_CAPI_BEGIN();
195 LITE_ASSERT(tensor, "The tensor pass to LITE c_api is null");
196 LITE_ASSERT(layout, "The layout ptr pass to LITE c_api is null");
197 *layout = convert_to_clayout(static_cast<lite::Tensor*>(tensor)->get_layout());
198 LITE_CAPI_END();
199}
200
201int LITE_get_tensor_device_type(const LiteTensor tensor, LiteDeviceType* device_type) {
202 LITE_CAPI_BEGIN();

Callers 6

start_callbackFunction · 0.85
start_callback_with_dataFunction · 0.85
finish_callbackFunction · 0.85
TESTFunction · 0.85
TESTFunction · 0.85

Calls 2

convert_to_clayoutFunction · 0.85
get_layoutMethod · 0.45

Tested by 6

start_callbackFunction · 0.68
start_callback_with_dataFunction · 0.68
finish_callbackFunction · 0.68
TESTFunction · 0.68
TESTFunction · 0.68