Searched refs:tfr_gen (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen_test.py | 29 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module as tfr_gen unknown
248 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_loc', [test_ops])
285 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_tensor', [test_ops])
363 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_control_flow', [test_ops])
424 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_tf_ops', [test_ops])
515 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_attrs', [test_ops])
556 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_shapes', [test_ops])
589 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_temp', [test_ops])
|
| D | tfr_gen.py | 1416 def tfr_gen(func, op_defs): function
1462 mlir_funcs = [tfr_gen(func, op_defs) for func in py_funcs]
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ |
| D | build_defs.bzl | 32 "//tensorflow/compiler/mlir/tfr:tfr_gen",
92 "//tensorflow/compiler/mlir/tfr:tfr_gen",
113 "//tensorflow/compiler/mlir/tfr:tfr_gen",
|
| D | define_op_template.py | 27 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module
|
| D | BUILD | 320 name = "tfr_gen",
321 srcs = ["python/tfr_gen.py"],
338 ":tfr_gen",
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/customization/ |
| D | ops_defs.py | 28 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | ops_defs.py | 30 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 30 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module
|