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

Function daliGetOperatorBackend

dali/c_api/c_api.cc:761–774  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

759}
760
761dali_backend_t daliGetOperatorBackend(daliPipelineHandle_t pipe_handle, const char *operator_name) {
762 dali::Pipeline* pipeline = (*pipe_handle)->pipeline.get();
763 auto *node = pipeline->GetOperatorNode(operator_name);
764 switch (node->op_type) {
765 case dali::OpType::CPU:
766 return dali_backend_t::DALI_BACKEND_CPU;
767 case dali::OpType::GPU:
768 return dali_backend_t::DALI_BACKEND_GPU;
769 case dali::OpType::MIXED:
770 return dali_backend_t::DALI_BACKEND_MIXED;
771 default:
772 DALI_FAIL("Invalid operator type.");
773 }
774}
775
776void daliGetExecutorMetadata(daliPipelineHandle_t pipe_handle, daliExecutorMetadata **operator_meta,
777 size_t *operator_meta_num) {

Callers 1

TESTFunction · 0.85

Calls 2

GetOperatorNodeMethod · 0.80
getMethod · 0.45

Tested by 1

TESTFunction · 0.68