Fix fastpath benchmarks
PiperOrigin-RevId: 331621386 Change-Id: I403ffe7259ed939b7f6c361c87dc4ef6ed3c5449
This commit is contained in:
parent
f9c29484b7
commit
00f20cc833
@ -80,7 +80,7 @@ def c_tfe_py_fastpath_execute(a,
|
||||
assert ctx.executing_eagerly(
|
||||
), "The prototype doesn't contain C code for graph construction"
|
||||
try:
|
||||
return pywrap_tfe.TFE_Py_FastPathExecute(ctx, ctx._handle,
|
||||
return pywrap_tfe.TFE_Py_FastPathExecute(ctx,
|
||||
"MatMul", name,
|
||||
a, b, "transpose_a", transpose_a,
|
||||
"transpose_b", transpose_b)
|
||||
|
Loading…
Reference in New Issue
Block a user