Home
last modified time | relevance | path

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

1234567

/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…
49 %0 = "tf.opA"(%arg0) {_tpu_replicate = "replicate"} : (tensor<i1>) -> tensor<i1>
51 %2 = "tf.opC"(%0) {_tpu_replicate = "replicate"} : (tensor<i1>) -> tensor<i1>
52 …"tf.TPUReplicateMetadata"() {_tpu_replicate = "replicate", device = "device", num_replicas = 1, to…
[all …]
Dreplicate_to_island.mlir1 // RUN: tf-opt -split-input-file -verify-diagnostics %s -tf-replicate-to-island | FileCheck %s
4 // replicate.
11 tf_device.replicate {n = 2 : i32} {
27 // Tests devices are not remapped if no devices were defined in replicate.
32 tf_device.replicate {n = 2 : i32} {
52 // Tests devices are not remapped if device is not in replicate devices.
57 tf_device.replicate {n = 2 : i32, devices = {CORE_0 = ["/CPU:0", "/GPU:1"]}} {
77 // Tests devices are remapped if device is in replicate devices.
82 tf_device.replicate {n = 2 : i32, devices = {CORE_0 = ["/CPU:0", "/GPU:1"]}} {
102 // Tests replicate with control dependency output has each expanded replica
[all …]
Dtf_device_ops.mlir34 tf_device.replicate {n = 2 : i32} {
38 // CHECK: tf_device.replicate
45 tf_device.replicate {n = 2 : i32} {
51 // CHECK: tf_device.replicate
70 …tf_device.replicate([%0, %1, %2] as %input0: tensor<*xi1>, %9 as %input1: tensor<*xi8>, %10 as %in…
87 // CHECK: tf_device.replicate
100 tf_device.replicate {n = 2 : i32, operand_segment_sizes = dense<[0, 0]> : vector<2xi32>} {
104 // CHECK: tf_device.replicate
113 %result:4 = tf_device.replicate([%arg0, %arg1] as %input0: tensor<*xf32>) {n = 2 : i32} {
118 // CHECK: tf_device.replicate
[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.cc146 TF::WhileRegionOp while_op, tf_device::ReplicateOp replicate, in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping() argument
154 replicate->getAttrOfType<ArrayAttr>(kMirroredVariableIndicesAttr); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
166 if (!block_arg || block_arg.getOwner() != &replicate.GetBody()) continue; in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
180 int64_t num_replicas = replicate.n(); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
198 const auto& block_arg = replicate.GetBody().getArgument(replicate_arg); in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
201 if (replicate.IsReplicatedBlockArgument(block_arg)) { in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
215 replicate.GetReplicaOperandForBlockArgument(block_arg, i), in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
231 return user != replicate && skipped_identities.count(user) == 0; in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
271 tf_device::ReplicateOp replicate, in AddInputsToReplicateOp() argument
275 int64_t num_replicas = replicate.n(); 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-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.
/external/llvm-project/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-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 …]

1234567