Commit 29831cdc authored by Shaba Abhiram's avatar Shaba Abhiram
Browse files

[BugFix] Fixes #20983 - Store py_func dtypes for correct conversion

1.  Implement wrapper class `PythonFunc` to store Tout list
    for appropriate dtype conversion from tensorflow types to
    numpy types using the `dtype.as_numpy_dtype` member.
2.  Adjust BUILD file to include py_test of the script_ops_test.
3.  Add script_ops_test to illustrate broken behavior (see
    `test_convert_empty_list`). Run the specific test suite with:

      bazel test //tensorflow/python:script_ops_test

4.  `_convert` was being called from dataset_ops, update this to
    refer to the `PythonFunc` instead of the `FuncRegistry`.
5.  [Review Feedback] Move script_ops tests to py_func_test.py. Also
    removed the BUILD rules for the deleted test file.
parent 2279279f
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment