Searched refs:experimental_get_compiler_ir (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | def_function_xla_jit_test.py | 182 fn.experimental_get_compiler_ir(inputs, inputs)())
199 g.experimental_get_compiler_ir(inputs, inputs)())
669 f.experimental_get_compiler_ir(a, b)(stage='optimized_hlo'))
686 f.experimental_get_compiler_ir(a, b)('optimized_hlo'))
698 f.experimental_get_compiler_ir(a)()
709 fn.experimental_get_compiler_ir(x, a)()
729 f.experimental_get_compiler_ir(b=a, a=b)(stage='hlo'))
743 f.experimental_get_compiler_ir(a, b)(stage='optimized_hlo_dot'))
758 f.experimental_get_compiler_ir(a, b)(stage='optimized_hlo_dot'))
770 f.experimental_get_compiler_ir(l)())
[all …]
|
| D | def_function.py | 953 def experimental_get_compiler_ir(self, *args, **kwargs): member in Function
|
| /external/tensorflow/ |
| D | RELEASE.md | 675 * Adds `tf.function.experimental_get_compiler_ir` which returns compiler IR
|