Home
last modified time | relevance | path

Searched refs:replicate (Results 1 – 25 of 147) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_device_ops_invalid.mlir3 // Check that a replicate with 'n' attribute that is less than 2 is invalid.
5 tf_device.replicate([%arg0] as %input0: tensor<*xf32>) {n = 1 : i32} {
6 // expected-error@-1 {{'tf_device.replicate' expects 'n' to be at least 2, got 1}}
13 // Check that a replicate replicated inputs where operand sizes do not match
16 tf_device.replicate([%arg0, %arg0, %arg0] as %input0: tensor<*xf32>) {n = 2 : i32} {
17 // expected-error@-1 {{'tf_device.replicate' expects number of operands for replicated input 0 to b…
24 // Check that a replicate with incompatible operands and block argument type is
28 tf_device.replicate([%arg0, %arg0] as %input0: tensor<*xf32>) {n = 2 : i32} {
36 // Check that a replicate with multiple blocks in its region is invalid.
38 tf_device.replicate() {n = 2 : i32} {
[all …]
Dtpu_cluster_formation.mlir8 …%0 = "tf.opA"() {_tpu_replicate = "replicate", device = "device", name = "name"} : () -> tensor<i1>
9 …"tf.TPUReplicateMetadata"() {_tpu_replicate = "replicate", device = "device", num_replicas = 1, to…
15 // CHECK-NOT: _tpu_replicate = "replicate"
24 %0 = "tf.opA"() {_tpu_replicate = "replicate"} : () -> tensor<i1>
25 …"tf.TPUReplicateMetadata"() {_tpu_replicate = "replicate", device = "device", name = "name", num_r…
36 %0 = "tf.opA"() {_tpu_replicate = "replicate"} : () -> tensor<i1>
37 …"tf.TPUReplicateMetadata"() {_tpu_replicate = "replicate", device = "device", num_replicas = 1, to…
51 …"tf.TPUReplicateMetadata"() {_tpu_replicate = "replicate", device = "device", num_replicas = 1, to…
52 %3 = "tf.opA"(%arg0) {_tpu_replicate = "replicate"} : (tensor<i1>) -> tensor<i1>
54 %5 = "tf.opC"(%3) {_tpu_replicate = "replicate"} : (tensor<i1>) -> tensor<i1>
[all …]
Dreplicate_invariant_op_hoisting.mlir1 // RUN: tf-opt %s -tf-replicate-invariant-op-hoisting | FileCheck %s --dump-input=fail
6 %0:4 = tf_device.replicate([%arg0, %arg1] as %ri: tensor<*xf32>) {n = 2: i32} {
15 // CHECK: tf_device.replicate([%[[ARG_0]], %[[ARG_1]]] as %[[RI:[a-z0-9]*]]: tensor<*xf32>)
23 %0:2 = tf_device.replicate([%arg0, %arg0] as %ri: tensor<*xf32>) {n = 2: i32} {
31 // CHECK: tf_device.replicate
38 %0:6 = tf_device.replicate([%arg0, %arg1] as %ri: tensor<*x!tf.resource>) {n = 2: i32} {
48 // CHECK: tf_device.replicate([%[[ARG_0]], %[[ARG_1]]] as %[[RI:[a-z0-9]*]]: tensor<*x!tf.resource>)
58 %1:2 = tf_device.replicate([%arg0, %arg0] as %ri: tensor<*x!tf.resource>) {n = 2: i32} {
67 // CHECK: tf_device.replicate
74 %0:6 = tf_device.replicate([%arg0, %arg1] as %ri: tensor<*xf32>) {n = 2: i32} {
[all …]
/external/eigen/test/
Darray_replicate.cpp12 template<typename MatrixType> void replicate(const MatrixType& m) in replicate() function
41 VERIFY_IS_APPROX(x1, m1.replicate(f1,f2)); in replicate()
46 VERIFY_IS_APPROX(x2, (m2.template replicate<2,3>())); in replicate()
50 VERIFY_IS_APPROX(x2, (m2.template replicate<1,3>())); in replicate()
54 VERIFY_IS_APPROX(vx1+vx1, vx1+(m2.template replicate<3,1>())); in replicate()
56 vx1=m2+(m2.colwise().replicate(1)); in replicate()
59 VERIFY_IS_APPROX(m2.coeff(0), (m2.template replicate<3,1>().coeff(m2.rows()))); in replicate()
64 VERIFY_IS_APPROX(x2, v1.rowwise().replicate(f1)); in replicate()
69 VERIFY_IS_APPROX(vx1, v1.colwise().replicate(f2)); in replicate()
75 CALL_SUBTEST_1( replicate(Matrix<float, 1, 1>()) ); in test_array_replicate()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_variable_runtime_reformatting.cc141 TF::WhileOp while_op, tf_device::ReplicateOp replicate, in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() argument
146 replicate.getAttrOfType<ArrayAttr>(kMirroredVariableIndicesAttr); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
158 if (!block_arg || block_arg.getOwner() != &replicate.GetBody()) continue; in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
171 int64_t num_replicas = replicate.n().getLimitedValue(); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
200 SkipIdentity(replicate.getOperand(num_replicas * replicate_arg + i), in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
213 user != replicate; in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
246 tf_device::ReplicateOp AddInputsToReplicateOp(tf_device::ReplicateOp replicate, in AddInputsToReplicateOp() argument
249 int64_t num_replicas = replicate.n().getLimitedValue(); in AddInputsToReplicateOp()
255 for (auto arg : llvm::enumerate(replicate.GetBody().getArguments())) { in AddInputsToReplicateOp()
259 replicated_inputs.back().push_back(replicate.getOperand(j)); in AddInputsToReplicateOp()
[all …]
/external/llvm/test/CodeGen/SystemZ/
Dvec-const-10.ll5 ; Test a byte-granularity replicate with the lowest useful value.
13 ; Test a byte-granularity replicate with an arbitrary value.
21 ; Test a byte-granularity replicate with the highest useful value.
29 ; Test a halfword-granularity replicate with the lowest useful value.
37 ; Test a halfword-granularity replicate with an arbitrary value.
45 ; Test a halfword-granularity replicate with the highest useful value.
53 ; Test a word-granularity replicate with the lowest useful positive value.
61 ; Test a word-granularity replicate with the highest in-range value.
69 ; Test a word-granularity replicate with the next highest value.
78 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-12.ll5 ; Test a byte-granularity replicate with the lowest useful value.
13 ; Test a byte-granularity replicate with an arbitrary value.
21 ; Test a byte-granularity replicate with the highest useful value.
29 ; Test a halfword-granularity replicate with the lowest useful value.
37 ; Test a halfword-granularity replicate with an arbitrary value.
45 ; Test a halfword-granularity replicate with the highest useful value.
53 ; Test a word-granularity replicate with the lowest useful positive value.
61 ; Test a word-granularity replicate with the highest in-range value.
69 ; Test a word-granularity replicate with the next highest value.
78 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-09.ll5 ; Test a byte-granularity replicate with the lowest useful value.
13 ; Test a byte-granularity replicate with an arbitrary value.
21 ; Test a byte-granularity replicate with the highest useful value.
29 ; Test a halfword-granularity replicate with the lowest useful value.
37 ; Test a halfword-granularity replicate with an arbitrary value.
45 ; Test a halfword-granularity replicate with the highest useful value.
53 ; Test a word-granularity replicate with the lowest useful positive value.
61 ; Test a word-granularity replicate with the highest in-range value.
69 ; Test a word-granularity replicate with the next highest value.
78 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-08.ll5 ; Test a byte-granularity replicate with the lowest useful value.
14 ; Test a byte-granularity replicate with an arbitrary value.
23 ; Test a byte-granularity replicate with the highest useful value.
32 ; Test a halfword-granularity replicate with the lowest useful value.
41 ; Test a halfword-granularity replicate with an arbitrary value.
50 ; Test a halfword-granularity replicate with the highest useful value.
59 ; Test a word-granularity replicate with the lowest useful positive value.
68 ; Test a word-granularity replicate with the highest in-range value.
77 ; Test a word-granularity replicate with the next highest value.
87 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-11.ll5 ; Test a byte-granularity replicate with the lowest useful value.
14 ; Test a byte-granularity replicate with an arbitrary value.
23 ; Test a byte-granularity replicate with the highest useful value.
32 ; Test a halfword-granularity replicate with the lowest useful value.
41 ; Test a halfword-granularity replicate with an arbitrary value.
50 ; Test a halfword-granularity replicate with the highest useful value.
59 ; Test a word-granularity replicate with the lowest useful positive value.
68 ; Test a word-granularity replicate with the highest in-range value.
77 ; Test a word-granularity replicate with the next highest value.
87 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-16.ll5 ; Test a word-granularity replicate with the lowest value that cannot use
14 ; Test a word-granularity replicate that has the lower 17 bits set.
22 ; Test a word-granularity replicate that has the upper 15 bits set.
30 ; Test a word-granularity replicate that has middle bits set.
38 ; Test a word-granularity replicate with a wrap-around mask.
46 ; Test a doubleword-granularity replicate with the lowest value that cannot
55 ; Test a doubleword-granularity replicate that has the lower 22 bits set.
63 ; Test a doubleword-granularity replicate that has the upper 45 bits set.
71 ; Test a doubleword-granularity replicate that has middle bits set.
79 ; Test a doubleword-granularity replicate with a wrap-around mask.
Dvec-const-15.ll5 ; Test a word-granularity replicate with the lowest value that cannot use
14 ; Test a word-granularity replicate that has the lower 17 bits set.
22 ; Test a word-granularity replicate that has the upper 15 bits set.
30 ; Test a word-granularity replicate that has middle bits set.
38 ; Test a word-granularity replicate with a wrap-around mask.
46 ; Test a doubleword-granularity replicate with the lowest value that cannot
55 ; Test a doubleword-granularity replicate that has the lower 22 bits set.
63 ; Test a doubleword-granularity replicate that has the upper 45 bits set.
71 ; Test a doubleword-granularity replicate that has middle bits set.
79 ; Test a doubleword-granularity replicate with a wrap-around mask.
Dvec-const-18.ll5 ; Test a word-granularity replicate with the lowest value that cannot use
14 ; Test a word-granularity replicate that has the lower 17 bits set.
22 ; Test a word-granularity replicate that has the upper 15 bits set.
30 ; Test a word-granularity replicate that has middle bits set.
38 ; Test a word-granularity replicate with a wrap-around mask.
46 ; Test a doubleword-granularity replicate with the lowest value that cannot
55 ; Test a doubleword-granularity replicate that has the lower 22 bits set.
63 ; Test a doubleword-granularity replicate that has the upper 45 bits set.
71 ; Test a doubleword-granularity replicate that has middle bits set.
79 ; Test a doubleword-granularity replicate with a wrap-around mask.
Dvec-const-17.ll5 ; Test a word-granularity replicate with the lowest value that cannot use
15 ; Test a word-granularity replicate that has the lower 17 bits set.
24 ; Test a word-granularity replicate that has the upper 15 bits set.
33 ; Test a word-granularity replicate that has middle bits set.
42 ; Test a word-granularity replicate with a wrap-around mask.
51 ; Test a doubleword-granularity replicate with the lowest value that cannot
61 ; Test a doubleword-granularity replicate that has the lower 22 bits set.
70 ; Test a doubleword-granularity replicate that has the upper 45 bits set.
79 ; Test a doubleword-granularity replicate that has middle bits set.
88 ; Test a doubleword-granularity replicate with a wrap-around mask.
Dvec-const-07.ll5 ; Test a byte-granularity replicate with the lowest useful value.
16 ; Test a byte-granularity replicate with an arbitrary value.
27 ; Test a byte-granularity replicate with the highest useful value.
38 ; Test a halfword-granularity replicate with the lowest useful value.
49 ; Test a halfword-granularity replicate with an arbitrary value.
60 ; Test a halfword-granularity replicate with the highest useful value.
71 ; Test a word-granularity replicate with the lowest useful positive value.
82 ; Test a word-granularity replicate with the highest in-range value.
93 ; Test a word-granularity replicate with the next highest value.
105 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
/external/swiftshader/third_party/llvm-7.0/llvm/test/CodeGen/SystemZ/
Dvec-const-10.ll5 ; Test a byte-granularity replicate with the lowest useful value.
13 ; Test a byte-granularity replicate with an arbitrary value.
21 ; Test a byte-granularity replicate with the highest useful value.
29 ; Test a halfword-granularity replicate with the lowest useful value.
37 ; Test a halfword-granularity replicate with an arbitrary value.
45 ; Test a halfword-granularity replicate with the highest useful value.
53 ; Test a word-granularity replicate with the lowest useful positive value.
61 ; Test a word-granularity replicate with the highest in-range value.
69 ; Test a word-granularity replicate with the next highest value.
78 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-09.ll5 ; Test a byte-granularity replicate with the lowest useful value.
13 ; Test a byte-granularity replicate with an arbitrary value.
21 ; Test a byte-granularity replicate with the highest useful value.
29 ; Test a halfword-granularity replicate with the lowest useful value.
37 ; Test a halfword-granularity replicate with an arbitrary value.
45 ; Test a halfword-granularity replicate with the highest useful value.
53 ; Test a word-granularity replicate with the lowest useful positive value.
61 ; Test a word-granularity replicate with the highest in-range value.
69 ; Test a word-granularity replicate with the next highest value.
78 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-12.ll5 ; Test a byte-granularity replicate with the lowest useful value.
13 ; Test a byte-granularity replicate with an arbitrary value.
21 ; Test a byte-granularity replicate with the highest useful value.
29 ; Test a halfword-granularity replicate with the lowest useful value.
37 ; Test a halfword-granularity replicate with an arbitrary value.
45 ; Test a halfword-granularity replicate with the highest useful value.
53 ; Test a word-granularity replicate with the lowest useful positive value.
61 ; Test a word-granularity replicate with the highest in-range value.
69 ; Test a word-granularity replicate with the next highest value.
78 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-11.ll5 ; Test a byte-granularity replicate with the lowest useful value.
14 ; Test a byte-granularity replicate with an arbitrary value.
23 ; Test a byte-granularity replicate with the highest useful value.
32 ; Test a halfword-granularity replicate with the lowest useful value.
41 ; Test a halfword-granularity replicate with an arbitrary value.
50 ; Test a halfword-granularity replicate with the highest useful value.
59 ; Test a word-granularity replicate with the lowest useful positive value.
68 ; Test a word-granularity replicate with the highest in-range value.
77 ; Test a word-granularity replicate with the next highest value.
87 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-08.ll5 ; Test a byte-granularity replicate with the lowest useful value.
14 ; Test a byte-granularity replicate with an arbitrary value.
23 ; Test a byte-granularity replicate with the highest useful value.
32 ; Test a halfword-granularity replicate with the lowest useful value.
41 ; Test a halfword-granularity replicate with an arbitrary value.
50 ; Test a halfword-granularity replicate with the highest useful value.
59 ; Test a word-granularity replicate with the lowest useful positive value.
68 ; Test a word-granularity replicate with the highest in-range value.
77 ; Test a word-granularity replicate with the next highest value.
87 ; Test a word-granularity replicate with the lowest in-range value.
[all …]
Dvec-const-16.ll5 ; Test a word-granularity replicate with the lowest value that cannot use
14 ; Test a word-granularity replicate that has the lower 17 bits set.
22 ; Test a word-granularity replicate that has the upper 15 bits set.
30 ; Test a word-granularity replicate that has middle bits set.
38 ; Test a word-granularity replicate with a wrap-around mask.
46 ; Test a doubleword-granularity replicate with the lowest value that cannot
55 ; Test a doubleword-granularity replicate that has the lower 22 bits set.
63 ; Test a doubleword-granularity replicate that has the upper 45 bits set.
71 ; Test a doubleword-granularity replicate that has middle bits set.
79 ; Test a doubleword-granularity replicate with a wrap-around mask.
Dvec-const-15.ll5 ; Test a word-granularity replicate with the lowest value that cannot use
14 ; Test a word-granularity replicate that has the lower 17 bits set.
22 ; Test a word-granularity replicate that has the upper 15 bits set.
30 ; Test a word-granularity replicate that has middle bits set.
38 ; Test a word-granularity replicate with a wrap-around mask.
46 ; Test a doubleword-granularity replicate with the lowest value that cannot
55 ; Test a doubleword-granularity replicate that has the lower 22 bits set.
63 ; Test a doubleword-granularity replicate that has the upper 45 bits set.
71 ; Test a doubleword-granularity replicate that has middle bits set.
79 ; Test a doubleword-granularity replicate with a wrap-around mask.
Dvec-const-18.ll5 ; Test a word-granularity replicate with the lowest value that cannot use
14 ; Test a word-granularity replicate that has the lower 17 bits set.
22 ; Test a word-granularity replicate that has the upper 15 bits set.
30 ; Test a word-granularity replicate that has middle bits set.
38 ; Test a word-granularity replicate with a wrap-around mask.
46 ; Test a doubleword-granularity replicate with the lowest value that cannot
55 ; Test a doubleword-granularity replicate that has the lower 22 bits set.
63 ; Test a doubleword-granularity replicate that has the upper 45 bits set.
71 ; Test a doubleword-granularity replicate that has middle bits set.
79 ; Test a doubleword-granularity replicate with a wrap-around mask.
Dvec-const-17.ll5 ; Test a word-granularity replicate with the lowest value that cannot use
15 ; Test a word-granularity replicate that has the lower 17 bits set.
24 ; Test a word-granularity replicate that has the upper 15 bits set.
33 ; Test a word-granularity replicate that has middle bits set.
42 ; Test a word-granularity replicate with a wrap-around mask.
51 ; Test a doubleword-granularity replicate with the lowest value that cannot
61 ; Test a doubleword-granularity replicate that has the lower 22 bits set.
70 ; Test a doubleword-granularity replicate that has the upper 45 bits set.
79 ; Test a doubleword-granularity replicate that has middle bits set.
88 ; Test a doubleword-granularity replicate with a wrap-around mask.
Dvec-const-07.ll5 ; Test a byte-granularity replicate with the lowest useful value.
16 ; Test a byte-granularity replicate with an arbitrary value.
27 ; Test a byte-granularity replicate with the highest useful value.
38 ; Test a halfword-granularity replicate with the lowest useful value.
49 ; Test a halfword-granularity replicate with an arbitrary value.
60 ; Test a halfword-granularity replicate with the highest useful value.
71 ; Test a word-granularity replicate with the lowest useful positive value.
82 ; Test a word-granularity replicate with the highest in-range value.
93 ; Test a word-granularity replicate with the next highest value.
105 ; Test a word-granularity replicate with the lowest in-range value.
[all …]

123456