Home
last modified time | relevance | path

Searched refs:experimental_get_compiler_ir (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dlight_outside_compilation_test.py52 compiled_f.experimental_get_compiler_ir(z)('hlo'), r"""
69 compiled_f.experimental_get_compiler_ir()()
82 hlo = compiled_f.experimental_get_compiler_ir(shape)('hlo_no_metadata')
103 hlo = compiled_f.experimental_get_compiler_ir(z)('hlo_no_metadata')
127 compiled_f.experimental_get_compiler_ir(z)()
139 hlo = compiled_f.experimental_get_compiler_ir(z)('hlo_no_metadata')
160 hlo = compiled_f.experimental_get_compiler_ir(z, 5)('hlo_no_metadata')
181 hlo = compiled_f.experimental_get_compiler_ir(z)()
203 hlo = compiled_f.experimental_get_compiler_ir(
/external/tensorflow/tensorflow/python/eager/
Ddef_function_xla_jit_test.py188 hlo_str = fn.experimental_get_compiler_ir(inputs, inputs)()
210 hlo_str = fn.experimental_get_compiler_ir(inputs)()
224 fn.experimental_get_compiler_ir(inputs, inputs)())
241 g.experimental_get_compiler_ir(inputs, inputs)())
818 f.experimental_get_compiler_ir(a, b)(stage='optimized_hlo'))
835 f.experimental_get_compiler_ir(a, b)('optimized_hlo'))
847 f.experimental_get_compiler_ir(a)()
858 fn.experimental_get_compiler_ir(x, a)()
878 f.experimental_get_compiler_ir(b=a, a=b)(stage='hlo'))
892 f.experimental_get_compiler_ir(a, b)(stage='optimized_hlo_dot'))
[all …]
Dcontext_test.py203 result = test_func.experimental_get_compiler_ir(a)(stage=stage)
Ddef_function.py996 def experimental_get_compiler_ir(self, *args, **kwargs): member in Function
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.types.experimental.-generic-function.pbtxt10 name: "experimental_get_compiler_ir"
Dtensorflow.__internal__.function.-function.pbtxt29 name: "experimental_get_compiler_ir"
/external/tensorflow/tensorflow/compiler/tests/
Dxla_custom_call_ops_test.py43 hlo = compiled_f.experimental_get_compiler_ir(x, y)(stage='hlo')
Dcast_ops_test.py49 hlo = compiled_f.experimental_get_compiler_ir(x)(stage='hlo')
/external/tensorflow/tensorflow/python/types/
Dcore.py175 def experimental_get_compiler_ir(self, *args, **kwargs): member in GenericFunction
/external/tensorflow/tensorflow/python/ops/memory_tests/
DBUILD40 # TODO(b/238830423): This test uses experimental_get_compiler_ir, which is
Dcustom_gradient_memory_test.py148 hlo_proto_serialized = test_func.experimental_get_compiler_ir(a)(
/external/tensorflow/tensorflow/compiler/xla/g3doc/tutorials/
Djit_compile.ipynb253 "print(train_mnist.experimental_get_compiler_ir(images, labels)(stage='hlo'))"
/external/tensorflow/
DRELEASE.md4845 * Adds `tf.function.experimental_get_compiler_ir` which returns compiler IR