Searched refs:AutoMixedPrecisionImpl (Results 1 – 1 of 1) sorted by relevance
926 class AutoMixedPrecisionImpl { class928 AutoMixedPrecisionImpl(Cluster* cluster, in AutoMixedPrecisionImpl() function in tensorflow::grappler::__anon69ea7d4c0111::AutoMixedPrecisionImpl1018 NodeDef AutoMixedPrecisionImpl::BuildCastNode( in BuildCastNode()1039 bool AutoMixedPrecisionImpl::NodeHasF16KernelForTypeAttr( in NodeHasF16KernelForTypeAttr()1052 Status AutoMixedPrecisionImpl::PrintDebugLogs(bool preop, size_t timestamp) { in PrintDebugLogs()1106 void AutoMixedPrecisionImpl::LogSkippedNode(const NodeDef& node) const { in LogSkippedNode()1114 bool AutoMixedPrecisionImpl::MustPreserve(const NodeDef& node) const { in MustPreserve()1118 bool AutoMixedPrecisionImpl::IsOnDevice(const NodeDef& node, in IsOnDevice()1162 bool AutoMixedPrecisionImpl::IsOnSuitableGPUArch(const NodeDef& node) const { in IsOnSuitableGPUArch()1166 bool AutoMixedPrecisionImpl::ShouldProcess(const NodeDef& node) const { in ShouldProcess()[all …]