Searched refs:to_f16 (Results 1 – 1 of 1) sorted by relevance
994 NodeDef BuildCastNode(const MutableGraphView::OutputPort& src, bool to_f16,1019 const MutableGraphView::OutputPort& src, bool to_f16, in BuildCastNode() argument1021 DataType src_type = to_f16 ? DT_FLOAT : target_dtype_; in BuildCastNode()1022 DataType dst_type = to_f16 ? target_dtype_ : DT_FLOAT; in BuildCastNode()1024 !to_f16 ? kCastToFp32 in BuildCastNode()1938 bool to_f16 = dst_is_allow; in ChangeTypeAttrsAndAddCasts() local1940 << (to_f16 ? DataTypeString(target_dtype_) : "DT_FLOAT") in ChangeTypeAttrsAndAddCasts()1944 BuildCastNode(src, to_f16, src.node->device())); in ChangeTypeAttrsAndAddCasts()1945 if (to_f16 && !IsConstant(*node) && !IsVariable(*node) && in ChangeTypeAttrsAndAddCasts()