Searched refs:IsOnDevice (Results 1 – 1 of 1) sorted by relevance
961 bool IsOnDevice(const NodeDef& node, const string& device_type) const;1118 bool AutoMixedPrecisionImpl::IsOnDevice(const NodeDef& node, in IsOnDevice() function in tensorflow::grappler::__anon69ea7d4c0111::AutoMixedPrecisionImpl1284 !MustPreserve(node) && IsOnDevice(node, DEVICE_GPU) && in Optimize()1288 should_process = !MustPreserve(node) && IsOnDevice(node, DEVICE_CPU); in Optimize()