GraphBuilder.convert_dtype()#

GraphBuilder.convert_dtype(from_dtype, to_dtype)[source]#

Tries to convert the node values in the graph to the specified data type.

Works for nodes whose value is an array or pytree. Nodes whose value is of another type are silently ignored. :rtype: GraphBuilder