Searched defs:SetTensorDeviceInfo (Results 1 – 2 of 2) sorted by relevance
144 void SetTensorDeviceInfo(const kernel::KernelBuildInfo &selected_kernel_info, const CNodePtr &kerne… in SetTensorDeviceInfo() function
432 void SetTensorDeviceInfo(const CNodePtr &kernel_node) { in SetTensorDeviceInfo() function