MCPcopy Create free account
hub / github.com/NVIDIA/DALI / CreateTensor

Function CreateTensor

dali/c_api_2/data_objects_test.cc:474–488  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

472
473
474inline auto CreateTensor(daliBufferPlacement_t placement) {
475 daliTensor_h handle;
476 auto err = daliTensorCreate(&handle, placement);
477 switch (err) {
478 case DALI_SUCCESS:
479 break;
480 case DALI_ERROR_OUT_OF_MEMORY:
481 throw std::bad_alloc();
482 case DALI_ERROR_INVALID_ARGUMENT:
483 throw std::invalid_argument(daliGetLastErrorMessage());
484 default:
485 throw std::runtime_error(daliGetLastErrorMessage());
486 }
487 return dali::c_api::TensorHandle(handle);
488}
489
490void TestTensorResize(daliStorageDevice_t storage_device) {
491 daliBufferPlacement_t placement{};

Callers 2

TestTensorResizeFunction · 0.85
TESTFunction · 0.85

Calls 2

daliTensorCreateFunction · 0.85
daliGetLastErrorMessageFunction · 0.85

Tested by

no test coverage detected