Searched refs:training_attr (Results 1 – 1 of 1) sorted by relevance
558 const auto* training_attr = node_view->GetAttr(kIsTraining); in IsCpuCompatibleDataType() local559 if (training_attr != nullptr && training_attr->b()) return false; in IsCpuCompatibleDataType()