Commit01bd2dc

authored

fix documentation (#102)

1 parentac4acc6 commit01bd2dcCopy full SHA for 01bd2dc

File tree

-1

lines changed

-1

lines changed

Lines changed: 2 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,8 @@`
`17`	`17`
`18`	`18`	`def_cast_like(x,y,saturate):`
`19`	`19`	`ifbfloat16isNone:`
`20`		`-return (cast_to(x,y.dtype,saturate),)`
	`20`	`+to=np_dtype_to_tensor_dtype(y.dtype)`
	`21`	`+return (cast_to(x,to,saturate),)`
`21`	`22`	`ify.dtype==bfloat16andy.dtype.descr[0][0]=="bfloat16":`
`22`	`23`	`# np.uint16 == np.uint16 is True as well as np.uint16 == bfloat16`
`23`	`24`	`to=TensorProto.BFLOAT16`

Comments

(0)