(a,
b,
transpose_a=False,
transpose_b=False,
name=None)
| 64 | |
| 65 | |
| 66 | def c_tfe_py_fastpath_execute(a, |
| 67 | b, |
| 68 | transpose_a=False, |
| 69 | transpose_b=False, |
| 70 | name=None): |
| 71 | ctx = context.context() |
| 72 | assert ctx.executing_eagerly( |
| 73 | ), "The prototype doesn't contain C code for graph construction" |
| 74 | try: |
| 75 | return pywrap_tensorflow.TFE_Py_FastPathExecute( |
| 76 | ctx._handle, ctx.device_name, "MatMul", name, |
| 77 | ctx._post_execution_callbacks, a, b, "transpose_a", transpose_a, |
| 78 | "transpose_b", transpose_b) |
| 79 | except core._NotOkStatusException as e: |
| 80 | if name is not None: |
| 81 | message = e.message + " name: " + name |
| 82 | else: |
| 83 | message = e.message |
| 84 | six.raise_from(core._status_to_exception(e.code, message), None) |
| 85 | |
| 86 | |
| 87 | class SubclassedKerasModel(keras.Model): |
no test coverage detected