Searched refs:TryFindKernelDef (Results 1 – 1 of 1) sorted by relevance
71 Status TryFindKernelDef(const std::vector<DeviceType>& devices, in TryFindKernelDef() function158 s = TryFindKernelDef({node.device().c_str(), DEVICE_GPU, DEVICE_CPU}, node, in IsNodeOutputPortHostFriendly()195 s = internal::TryFindKernelDef( in IsNodeInputPortHostFriendly()234 Status s = TryFindKernelDef({DEVICE_CPU}, node, nullptr); in IsNodeHostCandidate()