Skip to content

Add use_cudnn option for fc function in nn.py #10389

@kexinzhao

Description

@kexinzhao

User may want to let the activation op in fc to use cudnn if the use_cudnn option is set to true.
Currently this only applies to the softmax op.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions