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
Loading
Please sign in to comment