Home
last modified time | relevance | path

Searched refs:island (Results 1 – 25 of 174) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dbreakup-islands.mlir12 %island:3 = tf_executor.island {
17 tf_executor.fetch %island#0, %island#1 : tensor<*xi32>, tensor<*xi32>
24 // CHECK: %[[ADD1:.*]], %[[ADD1_control:.*]] = tf_executor.island wraps "tf.Add"(%arg0, %arg1)
25 // CHECK: %[[ADD2:.*]], %[[ADD2_control:.*]] = tf_executor.island wraps "tf.Add"(%[[ADD1]], %ar…
33 %island1:3 = tf_executor.island {
38 %island2:3 = tf_executor.island(%island1#2) {
43 %island3 = tf_executor.island {
48 %island4 = tf_executor.island(%island1#2, %island2#2) {
60 // CHECK: %[[ADD1:.*]], %[[ADD1_control:.*]] = tf_executor.island wraps "tf.Add"(%arg0, %arg1)
61 // CHECK: %[[ADD2:.*]], %[[ADD2_control:.*]] = tf_executor.island wraps "tf.Add"(%[[ADD1]], %arg…
[all …]
Dexecutor_island_coarsening.mlir1 // RUN: tf-opt %s -tf-executor-island-coarsening | FileCheck %s
9 %1:2 = tf_executor.island {
13 %2:2 = tf_executor.island(%1#1) {
22 // CHECK: %[[ISLAND:.*]], %[[ISLAND_control:.*]] = tf_executor.island {
34 %1:2 = tf_executor.island {
38 %2:2 = tf_executor.island {
47 // CHECK: %[[ISLAND:.*]], %[[ISLAND_control:.*]] = tf_executor.island {
59 %1:2 = tf_executor.island {
62 %2:2 = tf_executor.island {
65 %3:2 = tf_executor.island {
[all …]
Dresource-device-inference.mlir13 // CHECK: tf_executor.island
14 %island = tf_executor.island {
34 tf_executor.fetch %island : !tf_executor.control
49 // CHECK: tf_executor.island
50 %island = tf_executor.island {
67 tf_executor.fetch %island : !tf_executor.control
77 // CHECK: tf_executor.island
78 %island = tf_executor.island {
89 tf_executor.fetch %island : !tf_executor.control
99 // CHECK: tf_executor.island
[all …]
Dparallel_execute_to_islands.mlir6 tf_executor.island() {
19 // CHECK: [[ISLAND_0_CTRL:%.+]] = tf_executor.island {
21 // CHECK: [[ISLAND_1_CTRL:%.+]] = tf_executor.island {
30 %1:2 = tf_executor.island {
34 %3:3 = tf_executor.island() {
49 // CHECK: [[INPUT_A:%.+]], {{%.+}} = tf_executor.island {
52 // CHECK: [[ISLAND_0_OUTPUT:%.+]], {{%.+}} = tf_executor.island {
55 // CHECK: [[ISLAND_1_OUTPUT:%.+]], {{%.+}} = tf_executor.island {
64 %1 = tf_executor.island {
67 %2:3 = tf_executor.island(%1) {
[all …]
Dgraph_pruning.mlir7 // CHECK: island
8 // CHECK: island
10 %1:2 = tf_executor.island {
13 %2:2 = tf_executor.island {
21 // Check that an unused island that doesn't contribute to the fetch is removed.
24 // CHECK: tf_executor.island
25 // CHECK-NOT: tf_executor.island
27 %1:2 = tf_executor.island {
32 %2:2 = tf_executor.island {
79 …%1:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<0> : tenso…
[all …]
Dreplicate_to_island.mlir1 // RUN: tf-opt -split-input-file -verify-diagnostics %s -tf-replicate-to-island | FileCheck %s
3 // Tests per replica island has same control operands as island holding
10 %3 = tf_executor.island(%1, %2) {
23 // CHECK: %{{.*}} = tf_executor.island(%[[CT_0]], %[[CT_1]])
24 // CHECK: %{{.*}} = tf_executor.island(%[[CT_0]], %[[CT_1]])
31 %0 = tf_executor.island {
56 %0 = tf_executor.island {
81 %0 = tf_executor.island {
103 // control pinned to a sink island.
107 %1 = tf_executor.island {
[all …]
Dexecutor_canonicalize.mlir4 // Test single graph with no outputs and one island is folded away.
9 %1:2 = tf_executor.island {
24 // Test single graph with some outputs and one island is folded away.
29 %1:4 = tf_executor.island {
51 %1:4 = tf_executor.island {
53 %3:4 = tf_executor.island {
79 %1:4 = tf_executor.island {
85 %6:3 = tf_executor.island {
96 // CHECK-NEXT: %[[ISLAND_0:.*]]:3, %{{.*}} = tf_executor.island {
101 // CHECK: %[[ISLAND_1:.*]]:2, %{{.*}} = tf_executor.island {
[all …]
Dgraph_pruning_preserve_ops.mlir10 %1 = tf_executor.island wraps "tf.NoOp"() : () -> ()
12 …%2 = tf_executor.island(%1) wraps "tf.TPUReplicateMetadata"() {allow_soft_placement = false, compu…
23 %1 = tf_executor.island wraps "tf.NoOp"() : () -> ()
25 %2, %3 = tf_executor.island(%1) wraps "tf.TPUCompilationResult"() : () -> tensor<!tf.string>
36 %1 = tf_executor.island wraps "tf.NoOp"() : () -> ()
38 %2, %3 = tf_executor.island wraps "tf.Identity"(%arg0) : (tensor<i1>) -> tensor<i1>
40 …%4, %5 = tf_executor.island(%1) wraps "tf.TPUReplicatedInput"(%2) {index = -1 : i64, is_mirrored_v…
51 %1 = tf_executor.island wraps "tf.NoOp"() : () -> ()
53 %2, %3 = tf_executor.island wraps "tf.Identity"(%arg0) : (tensor<i1>) -> tensor<i1>
55 %4, %5 = tf_executor.island(%1) wraps "tf.TPUReplicatedOutput"(%2) : (tensor<i1>) -> tensor<i1>
[all …]
Dside-effect-analysis-test.mlir15 // CHECK: tf_executor.island
16 %island:2 = tf_executor.island {
50 tf_executor.fetch %island#0 : tensor<32xf32>
70 // CHECK: tf_executor.island
71 %island = tf_executor.island {
107 tf_executor.fetch %island : !tf_executor.control
127 // CHECK: tf_executor.island
128 %island = tf_executor.island {
165 tf_executor.fetch %island : !tf_executor.control
185 // CHECK: tf_executor.island
[all …]
Dtf_executor_ops_printer.mlir3 // Tests printer for tf_executor.island "wraps" short form.
8 // CHECK: tf_executor.island wraps "tf.IdentityN"
9 %0:3 = tf_executor.island {
21 // CHECK: tf_executor.island
23 %0:3 = tf_executor.island {
35 // CHECK: tf_executor.island
37 %0:3 = tf_executor.island {
49 // CHECK: tf_executor.island
51 %0:3 = tf_executor.island {
63 // CHECK: tf_executor.island
[all …]
Dtf-executor-to-functional.mlir5 …rolTrigger' op is not supported for lifting out of tf_executor.graph, expected tf_executor.island}}
27 %control = tf_executor.island {
40 %output, %control = tf_executor.island {
55 %output0, %control0 = tf_executor.island {
59 %output1, %control1 = tf_executor.island {
75 %output0, %control0 = tf_executor.island {
79 %output1, %control1 = tf_executor.island(%control0) {
97 %output0_0, %output0_1, %control0 = tf_executor.island {
102 %output1_0, %output1_1, %control1 = tf_executor.island {
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_coarsening/
Dexecutor_tpuv1_island_coarsening.mlir1 // RUN: tf-opt %s -tf-executor-tpu-v1-island-coarsening | FileCheck %s
8 %1:2 = tf_executor.island {
12 %2:2 = tf_executor.island(%1#1) {
18 // CHECK: island
30 // CHECK: island
34 …%outputs, %control = tf_executor.island wraps "tf.Const"() {_tpu_replicate = "cluster", value = de…
35 …%outputs_0, %control_1 = tf_executor.island wraps "tf.Const"() {_tpu_replicate = "cluster", value …
36 …%outputs_3, %control_4 = tf_executor.island wraps "tf.AddV2"(%outputs, %outputs_0) {_tpu_replicate…
47 // CHECK: island
52 // CHECK: island wraps "tf.AddV2"
[all …]
Dwhile_op.mlir1 // RUN: tf-opt %s -tf-executor-tpu-v1-island-coarsening | FileCheck %s
9 …%1:2 = tf_executor.island wraps "tf.opA"(%arg0) {_tpu_replicate = "cluster"} : (tensor<i1>) -> ten…
10 …%2:2 = tf_executor.island wraps "tf.While"(%1#0) {name = "A", body = @while_body_with_cluster_attr…
11 …%3:2 = tf_executor.island wraps "tf.While"(%1#0) {name = "B", body = @while_body_with_wrong_cluste…
12 …%4:2 = tf_executor.island wraps "tf.While"(%1#0) {name = "C", body = @while_body_without_cluster_a…
13 …%6:2 = tf_executor.island wraps "tf.While"(%1#0) {name = "D", body = @while_body_without_cluster_a…
14 …%5:2 = tf_executor.island wraps "tf.While"(%1#0) {name = "E", body = @while_body_with_cluster_attr…
17 // CHECK-NOT: island
19 // CHECK-NOT: island
21 // CHECK-NOT: island
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/
Dend_to_end.mlir10 …%outputs, %control = tf_executor.island wraps "std.constant"() {value = dense<2.000000e+00> : tens…
11 …%outputs_0, %control_1 = tf_executor.island wraps "std.constant"() {value = dense<3.000000e+00> : …
12 …%control_2 = tf_executor.island wraps "tf.TPUReplicateMetadata"() {_tpu_replicate = "cluster", all…
13 …%outputs_3, %control_4 = tf_executor.island wraps "tf.Placeholder"() {device = "", dtype = "tfdtyp…
14 …%outputs_5, %control_6 = tf_executor.island wraps "tf.TPUReplicatedInput"(%outputs_3) {N = 1 : i64…
15 …%outputs_7, %control_8 = tf_executor.island wraps "tf.Identity"(%outputs_5) {T = "tfdtype$DT_FLOAT…
16 …%outputs_9, %control_10 = tf_executor.island wraps "tf.Mul"(%outputs_7, %outputs) {T = "tfdtype$DT…
17 …%outputs_11, %control_12 = tf_executor.island wraps "tf.Placeholder"() {device = "", dtype = "tfdt…
18 …%outputs_13, %control_14 = tf_executor.island wraps "tf.TPUReplicatedInput"(%outputs_11) {N = 1 : …
19 …%outputs_15, %control_16 = tf_executor.island wraps "tf.Identity"(%outputs_13) {T = "tfdtype$DT_FL…
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dexecutor_tpuv1_island_coarsening.cc126 IslandOp island = dyn_cast<IslandOp>(*op); in MergeIsland() local
127 if (!island || !island.WrapsSingleOp()) return success(); in MergeIsland()
128 Operation& wrapped_op = island.GetBody().front(); in MergeIsland()
138 << *island.getOperation() << "\n"); in MergeIsland()
141 SmallVector<IslandOp, 16> islands{island}; in MergeIsland()
142 SmallPtrSet<Operation*, 16> wrapped_ops{&island.GetBody().front()}; in MergeIsland()
208 IslandOp new_island = OpBuilder(island).create<IslandOp>( in MergeIsland()
209 island.getLoc(), result_types, in MergeIsland()
210 /*control=*/ControlType::get(island.getContext()), in MergeIsland()
211 /*controlInputs=*/island.getOperands()); in MergeIsland()
[all …]
Dexecutor_island_coarsening.cc68 llvm::Optional<IslandOp> GetOperandCandidateToMergeWith(IslandOp island) { in GetOperandCandidateToMergeWith() argument
69 Operation* graph_op = island->getParentOp(); in GetOperandCandidateToMergeWith()
73 for (Value input : island.controlInputs()) { in GetOperandCandidateToMergeWith()
80 island.walk([graph_op, &candidate](Operation* op) { in GetOperandCandidateToMergeWith()
97 llvm::Optional<IslandOp> GetResultCandidateToMergeWith(IslandOp island) { in GetResultCandidateToMergeWith() argument
98 Operation* graph_op = island->getParentOp(); in GetResultCandidateToMergeWith()
102 for (Operation* user : island.control().getUsers()) { in GetResultCandidateToMergeWith()
109 for (Value result : island.outputs()) { in GetResultCandidateToMergeWith()
221 auto move_inner_ops = [block, new_yield_op](IslandOp island) { in MoveInnerOpsToNewIsland() argument
222 auto& island_body = island.GetBody().getOperations(); in MoveInnerOpsToNewIsland()
[all …]
Dgraph_pruning.cc47 bool ShouldPreserveIsland(IslandOp island);
91 if (auto island = llvm::dyn_cast<IslandOp>(op)) { in VisitOp() local
93 island.body(), island.body(), [&](OpOperand* operand) { in VisitOp()
128 bool GraphPruningPass::ShouldPreserveIsland(IslandOp island) { in ShouldPreserveIsland() argument
129 auto result = island.walk([this](Operation* inner_op) { in ShouldPreserveIsland()
151 auto island = llvm::dyn_cast<IslandOp>(op); in PruneGraph() local
152 if (!island) continue; in PruneGraph()
153 if (ShouldPreserveIsland(island)) { in PruneGraph()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dbreakup-islands.cc81 if (auto island = dyn_cast<tf_executor::IslandOp>(&item)) { in runOnFunction() local
82 BreakUpIsland(island, side_effect_analysis, &new_control_inputs); in runOnFunction()
129 void PopulateEmptyIsland(tf_executor::IslandOp island) { in PopulateEmptyIsland() argument
130 OpBuilder builder(&island.GetBody(), island.GetBody().begin()); in PopulateEmptyIsland()
131 tf_executor::YieldOp yield = island.GetYield(); in PopulateEmptyIsland()
133 builder.create<TF::NoOp>(island.getLoc(), TypeRange{}, ValueRange{}); in PopulateEmptyIsland()
136 auto identity = builder.create<TF::IdentityOp>(island.getLoc(), in PopulateEmptyIsland()
141 island.getLoc(), yield.getOperandTypes(), yield.getOperands()); in PopulateEmptyIsland()
155 auto island = builder.create<tf_executor::IslandOp>( in CreateIsland() local
157 island.body().push_back(new Block); in CreateIsland()
[all …]
Dtf_functional_to_executor.cc76 auto island = builder.create<tf_executor::IslandOp>( in runOnFunction() local
80 ValueRange to_fetch = island.getResults(); in runOnFunction()
87 island.body().push_back(new Block); in runOnFunction()
88 island.body().front().getOperations().splice( in runOnFunction()
89 island.body().front().begin(), body.getOperations(), copy_range.begin(), in runOnFunction()
91 builder.setInsertionPointToEnd(&island.body().front()); in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dcase.mlir6 …%outputs, %control = tf_executor.island wraps "tf.Const"() {device = "", value = dense<1> : tensor…
7 …%outputs_0, %control_1 = tf_executor.island wraps "tf.Const"() {device = "", value = dense<0> : te…
8 …%outputs_2, %control_3 = tf_executor.island wraps "tf.Case"(%outputs_0, %outputs) {Tin = [i32], To…
9 …%outputs_4, %control_5 = tf_executor.island wraps "tf.Identity"(%outputs_2) {device = ""} : (tenso…
10 …%outputs_6, %control_7 = tf_executor.island wraps "tf.Case"(%outputs_0, %outputs) {Tin = [i32], To…
18 …%outputs, %control = tf_executor.island wraps "tf.Const"() {device = "", value = dense<1> : tensor…
19 …%outputs_0, %control_1 = tf_executor.island wraps "tf.AddV2"(%arg0, %outputs) {device = ""} : (ten…
27 …%outputs, %control = tf_executor.island wraps "tf.Const"() {device = "", value = dense<2> : tensor…
28 …%outputs_0, %control_1 = tf_executor.island wraps "tf.AddV2"(%arg0, %outputs) {device = ""} : (ten…
Dswitchn.mlir6 %outputs, %control = "tf_executor.island"() ( {
11 %outputs_2, %control_3 = "tf_executor.island"() ( {
15 %outputs_4, %control_5 = "tf_executor.island"(%control_3) ( {
19 %outputs_6, %control_7 = "tf_executor.island"() ( {
23 %outputs_8, %control_9 = "tf_executor.island"(%control_7) ( {
27 %outputs_10, %control_11 = "tf_executor.island"() ( {
31 %outputs_12, %control_13 = "tf_executor.island"(%control_11) ( {
35 %outputs_14, %control_15 = "tf_executor.island"() ( {
40 %outputs_18, %control_19 = "tf_executor.island"() ( {
44 %outputs_20, %control_21 = "tf_executor.island"() ( {
[all …]
Dlegalized_name.mlir6 …%0:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<0> : tenso…
8 …%1:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<1> : tenso…
10 …%2:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<2> : tenso…
12 …%3:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<2> : tenso…
14 …%4:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<3> : tenso…
16 …%5:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<3> : tenso…
18 …%6:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<3> : tenso…
Dunique_name.mlir6 …%0:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<0> : tenso…
8 …%1:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<1> : tenso…
10 …%2:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<2> : tenso…
12 …%3:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<2> : tenso…
14 …%4:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<3> : tenso…
16 …%5:2 = tf_executor.island wraps "tf.Const"() {dtype = "tfdtype$DT_INT32", value = dense<3> : tenso…
Dinvalid_input.mlir42 %0 = tf_executor.island {
50 // CHECK: Functions must be of a single Graph with single op Islands: tf_executor.island must perfe…
56 %0 = tf_executor.island {
66 // CHECK: Functions must be of a single Graph with single op Islands: tf_executor.island must perfe…
72 %0 = tf_executor.island {
81 // CHECK: Functions must be of a single Graph with single op Islands: tf_executor.island must perfe…
87 %0:3 = tf_executor.island {
96 // CHECK: Functions must be of a single Graph with single op Islands: tf_executor.island must perfe…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_outline_island/
Dexecutor_tpuv1_outline_tpu_island.mlir1 // RUN: tf-opt %s -tf-executor-tpu-v1-island-outlining | FileCheck %s
7 // CHECK: island
10 %1:2 = tf_executor.island {
15 %2:2 = tf_executor.island(%1#1) {
27 %1:2 = tf_executor.island {
31 // CHECK: island
33 // CHECK: island
36 %2:3 = tf_executor.island {

1234567