added an option to verify_grad(cast_to_output_type) to force the casting of the…
added an option to verify_grad(cast_to_output_type) to force the casting of the output to the same type as the op expect. This is needed to test the ConvOp grad in float32 type. Otherwise it upcast it to float64.
正在显示
请
注册
或者
登录
后发表评论