Skip to content

hhappy06/TensorFlowCustomOP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

#TensorFlowCustomOP My practice about building a custom op and gradient in tensorflow.
The op y = sin(w*x) and its grad are defined in the file sin_wx.cc.
You can run 'python test_my_layer.py' to test the op and gradient.

##The files:

  • gen-so.sh: compile script
  • sin_wx.cc: defination of op and its gradient
  • testing_my_layer.py: testing file

##Results: image

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published