Searched refs:SetTensorDeviceInfo (Results 1 – 4 of 4) sorted by relevance
32 void SetTensorDeviceInfo(const CNodePtr &kernel_node);
432 void SetTensorDeviceInfo(const CNodePtr &kernel_node) { in SetTensorDeviceInfo() function491 SetTensorDeviceInfo(kernel_node); in SetMatchedKernelInfo()531 SetTensorDeviceInfo(kernel_node); in SelectKernelInfo()
98 SetTensorDeviceInfo(cnode); in UpdateKernelInfo()478 SetTensorDeviceInfo(kernel_node); in SetGraphKernelInfo()
144 void SetTensorDeviceInfo(const kernel::KernelBuildInfo &selected_kernel_info, const CNodePtr &kerne… in SetTensorDeviceInfo() function341 SetTensorDeviceInfo(*graph_selected_info, kernel_node); in SetGraphKernelInfo()451 SetTensorDeviceInfo(*(builder->Build()), kernel_node); in SetKernelInfo()