Commit ac8e6739 authored by Christoph Boeddeker's avatar Christoph Boeddeker Committed by Shanqing Cai
Browse files

improve py_func (#15121) 

* improve py_func
 - relable inp to args and Tout to output_types
 - add arguments kwargs and output_shapes
 - allow args/kwargs/output_types/output_shapes to be a nested structure

* move new py_func to tf.contrib.framework
parent 322d3530
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment