Searched refs:castToFp16 (Results 1 – 1 of 1) sorted by relevance
591 NodeDef* castToFp16 = graph_def->add_node(); in MaybeRewriteCastToFp32() local593 (*castToFp16->mutable_attr())[attr_value.first] = attr_value.second; in MaybeRewriteCastToFp32()595 castToFp16->set_name(node_def->name() + "_split"); in MaybeRewriteCastToFp32()596 castToFp16->set_op("Cast"); in MaybeRewriteCastToFp32()597 castToFp16->set_device(node_def->device()); in MaybeRewriteCastToFp32()598 castToFp16->add_input(node_def->input(0)); in MaybeRewriteCastToFp32()599 (*castToFp16->mutable_attr())[kCastOutputTypeAttrName].set_type(DT_HALF); in MaybeRewriteCastToFp32()601 node_def->set_input(0, castToFp16->name() + ":0"); in MaybeRewriteCastToFp32()604 VLOG(2) << castToFp16->DebugString(); in MaybeRewriteCastToFp32()