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