diff options
author | Akshay Modi <nareshmodi@google.com> | 2018-02-06 13:53:56 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-06 13:59:51 -0800 |
commit | ee5a6641e3a2db7807a655bef21614e97b82c769 (patch) | |
tree | d49c5747f02ecc9126af08468b3d3377beb44795 /tensorflow/python/pywrap_tfe.i | |
parent | d55d0d863c4e705846fc61ba3af7569cca509d67 (diff) |
Make TFE_Py_FastpathExecute work for all types of ops
MatMul benchmarks:
entry {
name: "MicroBenchmarks.benchmark_gen_math_ops_matmul_2_by_2_CPU"
iters: 30000
wall_time: 11.580435435
extras {
key: "examples_per_sec"
value {
double_value: 86352.538781
}
}
}
entry {
name: "MicroBenchmarks.benchmark_tfe_py_fastpath_execute_matmul_2_by_2_CPU"
iters: 30000
wall_time: 7.02576637268
extras {
key: "examples_per_sec"
value {
double_value: 142333.227004
}
}
}
PiperOrigin-RevId: 184734289
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
-rw-r--r-- | tensorflow/python/pywrap_tfe.i | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 3f25311a83..50f481d29e 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -29,6 +29,7 @@ limitations under the License. %rename("%s") TFE_OpNameGetAttrType; %rename("%s") TFE_Py_InitEagerTensor; %rename("%s") TFE_Py_RegisterExceptionClass; +%rename("%s") TFE_Py_RegisterFallbackExceptionClass; %rename("%s") TFE_Py_Execute; %rename("%s") TFE_Py_FastPathExecute; %rename("%s") TFE_Py_UID; |