Home
last modified time | relevance | path

Searched refs:tfr (Results 1 – 25 of 52) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/tfr/tests/
Dops.mlir1 // RUN: tfr-opt %s -verify-diagnostics -split-input-file | tfr-opt | FileCheck %s
2 // RUN: tfr-opt %s -canonicalize -verify-diagnostics -split-input-file | FileCheck %s -check-prefix…
8 func private @tensor_type_noconstraint() -> !tfr.tensor
13 func private @tensor_type() -> !tfr.tensor<T>
18 func private @tensor_list_type_noconstraint() -> !tfr.tensor_list
23 func private @tensor_list_type_array_like() -> !tfr.tensor_list<[N, T]>
28 func private @tensor_list_type_tuple_like() -> !tfr.tensor_list<input_T>
33 func private @tensor_invalid_1() -> !tfr.tensor<[N, T>
38 func @tensor_invalid_2() -> !tfr.tensor<[N, T]
43 func @call_op(%arg0: !tfr.tensor<T>, %arg1: !tfr.tensor_list<TL>, %arg2: i32) -> !tfr.tensor<K> {
[all …]
Draise_to_tf.mlir1 // RUN: tfr-opt %s -tfr-raise-to-tf -verify-diagnostics -split-input-file | FileCheck %s
3 tfr.func @tf__risc_same_(!tfr.tensor<T>) -> !tfr.tensor<T> attributes {T}
4 tfr.func @tf__risc_concat_(!tfr.tensor_list<N, T>) -> !tfr.tensor<T> attributes {T, N}
5 tfr.func @tf__risc_split_(!tfr.tensor<T>, i32 {tfr.name="N"}) -> !tfr.tensor_list<N, T> attributes …
6 tfr.func @tf__risc_cast_(!tfr.tensor, !tfr.attr {tfr.name="K"}) -> !tfr.tensor<K> attributes {T, K}
7 tfr.func @tf__const_(!tfr.attr {tfr.name="value", tfr.type="tensor"},
8 !tfr.attr {tfr.name="K",tfr.type="dtype"}) -> !tfr.tensor<K> attributes {T, K}
12 %0 = "tfr.cast"(%arg0) : (tensor<1x2x3x4x!tf.string>) -> !tfr.tensor
13 %1 = tfr.call @tf__risc_same(%0) : (!tfr.tensor) -> !tfr.tensor
14 %2 = "tfr.cast"(%1) : (!tfr.tensor) -> tensor<1x2x3x4x!tf.string>
[all …]
Dend2end.mlir1 // RUN: tfr-opt %s -tfr-decompose -tfr-raise-to-tf -canonicalize -verify-diagnostics -split-input-f…
125 tfr.func @tf__my_identity(%value: !tfr.tensor) -> !tfr.tensor {
126 tfr.return %value : !tfr.tensor
129 tfr.func @tf__my_cast(%value: !tfr.tensor, %tout: !tfr.attr{tfr.name="Tout"}) -> !tfr.tensor {
130 %0 = tfr.call @tf__risc_cast(%value, %tout) : (!tfr.tensor, !tfr.attr) -> !tfr.tensor
131 tfr.return %0 : !tfr.tensor
134 tfr.func @tf__my_rsqrt(%value: !tfr.tensor) -> !tfr.tensor {
135 %1 = tfr.call @tf__risc_reciprocal(%value) : (!tfr.tensor) -> !tfr.tensor
136 %2 = tfr.call @tf__risc_sqrt(%1) : (!tfr.tensor) -> !tfr.tensor
137 tfr.return %2 : !tfr.tensor
[all …]
Ddecompose.mlir1 // RUN: tfr-opt %s -tfr-decompose -verify-diagnostics -split-input-file | FileCheck %s
4 tfr.func @tf__fake_no_op(%arg0: !tfr.tensor) -> !tfr.tensor {
5 tfr.return %arg0 : !tfr.tensor
7 // CHECK-NEXT: tfr.return %arg0 : !tfr.tensor
11 tfr.func @tf__intermediate(%arg0: !tfr.tensor) -> !tfr.tensor {
12 %0 = tfr.call @tf__risc(%arg0) : (!tfr.tensor) -> !tfr.tensor
13 tfr.return %0 : !tfr.tensor
15 // CHECK-NEXT: %[[id:.*]] = tfr.call @tf__risc(%arg0) : (!tfr.tensor) -> !tfr.tensor
16 // CHECK-NEXT: tfr.return %[[id]] : !tfr.tensor
20 tfr.func @tf__fused_n(
[all …]
Dcontrol_flow.mlir1 // RUN: tfr-opt %s -tfr-decompose -verify-diagnostics -split-input-file | FileCheck %s
3 tfr.func @tf__my_pack(%values: !tfr.tensor_list,
4 %n: i32 {tfr.name="N"},
5 %axis: i32 {tfr.name="axis"}) -> !tfr.tensor {
9 %v1 = tfr.get_element %values[%index] : (!tfr.tensor_list, index) -> !tfr.tensor
10 %temp = tfr.call @tf__expand_dims(%v1, %axis) : (!tfr.tensor, i32) -> !tfr.tensor
11 %res = scf.if %eq -> !tfr.tensor {
12 scf.yield %temp : !tfr.tensor
16 %reduce = scf.for %i = %step to %end step %step iter_args(%reduce_iter=%temp) -> !tfr.tensor {
17 %v = tfr.get_element %values[%i] : (!tfr.tensor_list, index) -> !tfr.tensor
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfr/resources/
Ddecomposition_lib.mlir1 // A test resource file which contains some pre-defined internal tfr.functions
2 // for decomposition and external tfr.functions for raising the decomposition
5 // All the tfr.func functions are supposed to be translated from the Python
7 // All the external tfr.func functions modeles the op signature defined by
10 tfr.func @tf__my_add_n(%values: !tfr.tensor_list,
11 %n: i64 {tfr.name="N"}) -> !tfr.tensor {
15 %v1 = tfr.get_element %values[%index] : (!tfr.tensor_list, index) -> !tfr.tensor
16 %res = scf.if %eq -> !tfr.tensor {
17 scf.yield %v1 : !tfr.tensor
21 %reduce = scf.for %i = %step to %end step %step iter_args(%reduce_iter=%v1) -> !tfr.tensor {
[all …]
DBUILD1 load("//tensorflow/compiler/mlir/tfr:build_defs.bzl", "gen_op_bindings")
14 "//learning/brain/experimental/mlir/tfr/...",
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.td34 let name = "tfr";
71 TFR_AllAttrTypes.predicate]>, "allowed tfr.call operand types">;
77 ArrayAttr.predicate]>, "allowed tfr.constant value"> {
86 "allowed tfr.call result types">;
88 // standard tensor type and tfr.tensor types can be casted to each other.
91 TF_Tensor.predicate]>, "single tensor or tfr.tensor type">;
97 TFR_AttrType.predicate]>, "single tfr.tensor or tensor element type">;
102 TFR_AttrType.predicate]>, "tfr.tensor_list or tfr.attr type">;
121 %2 = tfr.call @my_add(%0, %1) : (tfr.tensor, f32) -> tfr.tensor_list
124 Note that the operands of the `call` operation can only be with tfr.tensor,
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfr/
DBUILD8 load("//tensorflow/compiler/mlir/tfr:build_defs.bzl", "gen_op_libraries")
27 "//learning/brain/experimental/mlir/tfr/...",
71 name = "tfr",
107 ":tfr",
125 ":tfr",
145 name = "tfr-opt",
149 ":tfr",
176 "//tensorflow/compiler/mlir/tfr:tfr-opt",
188 ":tfr",
247 data = ["//tensorflow/compiler/mlir/tfr/resources:decomposition_lib"],
[all …]
Dbuild_defs.bzl31 "//tensorflow/compiler/mlir/tfr:op_reg_gen",
32 "//tensorflow/compiler/mlir/tfr:tfr_gen",
33 "//tensorflow/compiler/mlir/tfr:composite",
91 "//tensorflow/compiler/mlir/tfr:op_reg_gen",
92 "//tensorflow/compiler/mlir/tfr:tfr_gen",
93 "//tensorflow/compiler/mlir/tfr:composite",
112 "//tensorflow/compiler/mlir/tfr:op_reg_gen",
113 "//tensorflow/compiler/mlir/tfr:tfr_gen",
114 "//tensorflow/compiler/mlir/tfr:composite",
Ddefine_op_template.py25 from tensorflow.compiler.mlir.tfr.python.composite import Composite
26 from tensorflow.compiler.mlir.tfr.python.op_reg_gen import gen_register_op
27 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module
/external/libchrome/base/
Dcallback_unittest.cc130 TestForReentrancy tfr; in TEST_F() local
131 ASSERT_FALSE(tfr.cb.is_null()); in TEST_F()
132 ASSERT_FALSE(tfr.cb_already_run); in TEST_F()
133 ResetAndReturn(&tfr.cb).Run(); in TEST_F()
134 ASSERT_TRUE(tfr.cb.is_null()); in TEST_F()
135 ASSERT_TRUE(tfr.cb_already_run); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/customization/
DBUILD2 load("//tensorflow/compiler/mlir/tfr:build_defs.bzl", "gen_op_libraries")
15 "//tensorflow/compiler/mlir/tfr/...",
41 "//tensorflow/compiler/mlir/tfr:test_utils",
Dops_defs.py26 from tensorflow.compiler.mlir.tfr.python import composite
27 from tensorflow.compiler.mlir.tfr.python.op_reg_gen import gen_register_op
28 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/
DBUILD2 load("//tensorflow/compiler/mlir/tfr:build_defs.bzl", "gen_op_libraries")
15 "//tensorflow/compiler/mlir/tfr/...",
43 "//tensorflow/compiler/mlir/tfr:test_utils",
Dpad_ops_test.py24 from tensorflow.compiler.mlir.tfr.examples.pad import gen_pad_ops
25 from tensorflow.compiler.mlir.tfr.examples.pad import ops_defs
26 from tensorflow.compiler.mlir.tfr.python import test_utils
Dops_defs.py28 from tensorflow.compiler.mlir.tfr.python import composite
29 from tensorflow.compiler.mlir.tfr.python.op_reg_gen import gen_register_op
30 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dmnist_ops_test.py23 from tensorflow.compiler.mlir.tfr.examples.mnist import gen_mnist_ops
24 from tensorflow.compiler.mlir.tfr.examples.mnist import ops_defs
25 from tensorflow.compiler.mlir.tfr.python import test_utils
DBUILD2 load("//tensorflow/compiler/mlir/tfr:build_defs.bzl", "gen_op_libraries")
16 "//tensorflow/compiler/mlir/tfr/...",
44 "//tensorflow/compiler/mlir/tfr:test_utils",
Dops_defs.py28 from tensorflow.compiler.mlir.tfr.python import composite
29 from tensorflow.compiler.mlir.tfr.python.op_reg_gen import gen_register_op
30 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module
Dmnist_train.py25 from tensorflow.compiler.mlir.tfr.examples.mnist import gen_mnist_ops
26 from tensorflow.compiler.mlir.tfr.examples.mnist import ops_defs # pylint: disable=unused-import
/external/tensorflow/tensorflow/compiler/mlir/tfr/python/
Dop_reg_gen_test.py28 from tensorflow.compiler.mlir.tfr.python import composite
29 from tensorflow.compiler.mlir.tfr.python.op_reg_gen import gen_register_op
Dtfr_gen_test.py28 from tensorflow.compiler.mlir.tfr.python import composite
29 from tensorflow.compiler.mlir.tfr.python.tfr_gen import tfr_gen_from_module as tfr_gen
30 from tensorflow.compiler.mlir.tfr.resources import gen_test_ops as test_ops
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dtfr_decompose_ctx_test.cc92 test_ctx_ = tfr::TFRDecomposeContext::GetFromText(tfr_raw_text, &ctx_); in SetUp()
98 std::unique_ptr<tfr::TFRDecomposeContext> test_ctx_;
Dgraph_decompose_pass.h24 namespace tfr {

123