Home
last modified time | relevance | path

Searched refs:AddV2 (Results 1 – 25 of 73) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dgpu_op_add.cc20 GENERATE_AND_REGISTER_BINARY_GPU_KERNEL(AddV2, DT_HALF);
21 GENERATE_AND_REGISTER_BINARY_GPU_KERNEL(AddV2, DT_FLOAT);
22 GENERATE_AND_REGISTER_BINARY_GPU_KERNEL(AddV2, DT_DOUBLE);
23 GENERATE_AND_REGISTER_BINARY_GPU_KERNEL(AddV2, DT_INT64);
26 REGISTER_ALIASED_GPU_KERNEL(Add, AddV2, DT_HALF, DT_HALF);
27 REGISTER_ALIASED_GPU_KERNEL(Add, AddV2, DT_FLOAT, DT_FLOAT);
28 REGISTER_ALIASED_GPU_KERNEL(Add, AddV2, DT_DOUBLE, DT_DOUBLE);
29 REGISTER_ALIASED_GPU_KERNEL(Add, AddV2, DT_INT64, DT_INT64);
Dcpu_op_add.cc20 GENERATE_AND_REGISTER_BINARY_CPU_KERNEL(AddV2, DT_HALF);
21 GENERATE_AND_REGISTER_BINARY_CPU_KERNEL(AddV2, DT_FLOAT);
22 GENERATE_AND_REGISTER_BINARY_CPU_KERNEL(AddV2, DT_DOUBLE);
23 GENERATE_AND_REGISTER_BINARY_CPU_KERNEL(AddV2, DT_INT32);
24 GENERATE_AND_REGISTER_BINARY_CPU_KERNEL(AddV2, DT_INT64);
Dcpu_binary_ops_test.cc43 GENERATE_DEFAULT_TESTS(AddV2, /*test_name=*/Half, Eigen::half, Eigen::half,
45 GENERATE_DEFAULT_TESTS(AddV2, /*test_name=*/Float, float, float, baseline_add)
46 GENERATE_DEFAULT_TESTS(AddV2, /*test_name=*/Double, double, double,
48 GENERATE_DEFAULT_TESTS(AddV2, /*test_name=*/Int32, int32, int32, baseline_add)
49 GENERATE_DEFAULT_TESTS(AddV2, /*test_name=*/Int64, int64, int64, baseline_add)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_coarsening/
Dexecutor_tpuv1_island_coarsening.mlir33 // CHECK-NEXT: = "tf.AddV2"
36 …%outputs_3, %control_4 = tf_executor.island wraps "tf.AddV2"(%outputs, %outputs_0) {_tpu_replicate…
50 // CHECK-NEXT: = "tf.AddV2"
52 // CHECK: island wraps "tf.AddV2"
55 …%outputs_3, %control_4 = tf_executor.island wraps "tf.AddV2"(%outputs, %outputs_0) : (tensor<i32>,…
56 …%outputs_5, %control_6 = tf_executor.island wraps "tf.AddV2"(%outputs, %outputs_0) {_tpu_replicate…
71 // CHECK-NEXT: = "tf.AddV2"{{.*}}"cluster1"
72 // CHECK-NEXT: = "tf.AddV2"{{.*}}"cluster1"
76 // CHECK-NEXT: = "tf.AddV2"{{.*}}"cluster2"
77 // CHECK-NEXT: = "tf.AddV2"{{.*}}"cluster2"
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dconstant_op_device_assignment.mlir7 …// CHECK-NEXT: %[[RESULT_2:.*]] = "tf.AddV2"(%[[RESULT_1]], %[[RESULT_1]]) {device = "/job:worker/…
8 …// CHECK-NEXT: %[[RESULT_3:.*]] = "tf.AddV2"(%[[RESULT_0]], %[[RESULT_0]]) {device = "/job:worker/…
10 …%1 = "tf.AddV2"(%0, %0) {device = "/job:worker/replica:0/task:0/device:CPU:0"} : (tensor<f32>, ten…
11 …%2 = "tf.AddV2"(%0, %0) {device = "/job:worker/replica:0/task:0/device:CPU:1"} : (tensor<f32>, ten…
18 …// CHECK-NEXT: %[[RESULT_1:.*]] = "tf.AddV2"(%[[RESULT_0]], %[[RESULT_0]]) : (tensor<i64>, tensor<…
20 %1 = "tf.AddV2"(%0, %0) : (tensor<i64>, tensor<i64>) -> tensor<i64>
Dtf_device_replication_pass.mlir5 …// CHECK-NEXT: %[[RESULT_0:.*]] = "tf.AddV2"(%[[ARG_0]], %[[ARG_0]]) {device = "/job:worker/replic…
6 …// CHECK-NEXT: %[[RESULT_1:.*]] = "tf.AddV2"(%[[ARG_0]], %[[ARG_0]]) {device = "/job:worker/replic…
7 …// CHECK-NEXT: %[[RESULT_2:.*]] = "tf.AddV2"(%[[ARG_1]], %[[ARG_1]]) {device = "/job:worker/replic…
8 …// CHECK-NEXT: %[[RESULT_3:.*]] = "tf.AddV2"(%[[ARG_1]], %[[ARG_1]]) {device = "/job:worker/replic…
10 …%1 = "tf.AddV2"(%arg2, %arg2) {device = "TPU_REPLICATED_CORE_0"} : (tensor<i32>, tensor<i32>) -> t…
Dpromote_resources_to_args.mlir8 // CHECK: %[[ADD:[0-9]*]] = "tf.AddV2"(%arg1, %[[CONST:[0-9]*]])
14 %3 = "tf.AddV2"(%2, %0) : (tensor<f32>, tensor<f32>) -> tensor<f32>
40 // CHECK: %[[ADD1:[0-9]*]] = "tf.AddV2"(%arg1, %[[CONST:[0-9]*]])
41 // CHECK: %[[ADD2:[0-9]*]] = "tf.AddV2"(%[[ADD1]], %arg1)
48 %3 = "tf.AddV2"(%2, %0) : (tensor<f32>, tensor<f32>) -> tensor<f32>
51 %6 = "tf.AddV2"(%3, %5) : (tensor<f32>, tensor<f32>) -> tensor<f32>
63 // CHECK: %[[ADD1:[0-9]*]] = "tf.AddV2"(%arg1, %[[CONST:[0-9]*]])
64 // CHECK: %[[ADD2:[0-9]*]] = "tf.AddV2"(%[[ADD1]], %arg2)
71 %3 = "tf.AddV2"(%2, %0) : (tensor<f32>, tensor<f32>) -> tensor<f32>
74 %6 = "tf.AddV2"(%3, %5) : (tensor<f32>, tensor<f32>) -> tensor<f32>
[all …]
Dlayout_optimization_move_transposes_begin.mlir42 …// CHECK: %[[ADD:[0-9]*]] = "tf.AddV2"(%[[ARG0_TRANSPOSE]], %[[ARG1_TRANSPOSE]]) {{.*}} tensor<1x8…
45 %0 = "tf.AddV2"(%arg0, %arg1) : (tensor<1x4x4x8xf32>, tensor<1x4x4x8xf32>) -> tensor<1x4x4x8xf32>
58 // CHECK: %[[ADD:[0-9]*]] = "tf.AddV2"(%[[TANH]], %[[TANH]]) {{.*}} tensor<1x8x4x4xf32>
62 %1 = "tf.AddV2"(%0, %0) : (tensor<1x4x4x8xf32>, tensor<1x4x4x8xf32>) -> tensor<1x4x4x8xf32>
75 …%1 = "tf.AddV2"(%0, %arg1) {device = ""} : (tensor<8x64x1xf32>, tensor<8x64x64xf32>) -> tensor<8x6…
87 …// CHECK: %[[ADD:.*]] = "tf.AddV2"(%[[TRANSPOSE_1]], %[[TRANSPOSE_2]]) {device = ""} : (tensor<1x8…
95 …%0 = "tf.AddV2"(%arg0, %arg1) {device = ""} : (tensor<1x1x2x3xf32>, tensor<2x3xf32>) -> tensor<1x1…
101 …// CHECK: %[[ADD:.*]] = "tf.AddV2"(%arg0, %arg1) {device = ""} : (tensor<1x1x2x3xf32>, tensor<2x3x…
Dcluster_tf_ops_pass.mlir17 …%2 = "tf.AddV2"(%arg1, %arg1) {device = "/job:worker/replica:0/task:1/device:CPU:0"} : (tensor<i32…
23 // CHECK-NEXT: %[[RESULT_0:.*]] = "tf.AddV2"(%[[ARG_0]], %[[ARG_0]])
45 // CHECK-NEXT: %[[RESULT_1:.*]] = "tf.AddV2"(%[[ARG_0]], %[[RESULT_0]])
55 …%1 = "tf.AddV2"(%arg0, %0) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<i32…
59 …%4 = "tf.AddV2"(%3, %3) {device = "/job:worker/replica:0/task:1/device:CPU:0"} : (tensor<i32>, ten…
63 …%6 = "tf.AddV2"(%5, %5) {device = "/job:worker/replica:0/task:2/device:CPU:0"} : (tensor<i32>, ten…
71 // CHECK-NEXT: %[[RESULT_1:.*]] = "tf.AddV2"(%[[RESULT_0]], %[[RESULT_0]])
77 // CHECK-NEXT: %[[RESULT_1:.*]] = "tf.AddV2"(%[[RESULT_0]], %[[RESULT_0]])
Dtpu_compile_op_replication_pass.mlir11 …// CHECK-NEXT: %[[ADD_0:.*]] = "tf.AddV2"(%[[ARG_0]], %[[ARG_0]]) {device = "/job:worker/replica:0…
12 …%0 = "tf.AddV2"(%arg0, %arg0) {device = "/job:worker/replica:0/task:0/device:TPU:0"} : (tensor<i32…
15 …// CHECK-NEXT: %[[ADD_1:.*]] = "tf.AddV2"(%[[ARG_1]], %[[ARG_1]]) {device = "/job:worker/replica:0…
16 …%2 = "tf.AddV2"(%arg1, %arg1) {device = "/job:worker/replica:0/task:1/device:TPU:0"} : (tensor<i32…
Ddecompose_resource_ops.mlir13 // CHECK: "tf.AddV2"(%[[RES_READ_VAL]], %[[ONE]])
35 // CHECK: "tf.AddV2"(%[[RES_READ_VAL]], %[[ONE]])
104 // CHECK: [[ACCUM_NEW:%.*]] = "tf.AddV2"([[ACCUM_MOMENTUM]], [[GRAD]])
130 // CHECK: [[ACCUM_NEW:%.*]] = "tf.AddV2"([[ACCUM_MOMENTUM]], [[GRAD]])
135 // CHECK: [[DELTA:%.*]] = "tf.AddV2"([[GRAD_LR]], [[ACCUM_NEW_MOMENTUM_LR]])
164 // CHECK: %[[NEW_VAR:[0-9]*]] = "tf.AddV2"(%[[VAR]], %[[NEW_ACCUM]])
195 // CHECK: %[[NEW_VAR:[0-9]*]] = "tf.AddV2"(%[[VAR]], %[[NEW_DELTA]])
216 // CHECK: [[NEW_ACC:%.*]] = "tf.AddV2"([[OLD_ACC]], [[GRAD_SQUARE]]) : (tensor<*xf32>, tensor<f32>)…
219 // CHECK: [[DIVISOR:%.*]] = "tf.AddV2"([[SQRT]], [[EPSILON]]) : (tensor<*xf32>, tensor<f32>) -> ten…
243 …// CHECK: %[[ACCUM_NEW:.*]] = "tf.AddV2"(%[[ACCUM]], %[[GRAD_SQUARE]]) : (tensor<*xf32>, tensor<f3…
[all …]
Doptimize.mlir16 // CHECK-NEXT: %[[bias:.*]] = "tf.AddV2"(%[[conv]], %[[cst_0]])
26 …%1 = "tf.AddV2"(%0, %bias) {T = "tfdtype$DT_FLOAT"}: (tensor<256x8x7x16xf32>, tensor<16xf32>) -> t…
33 // CHECK-NEXT: %[[add:.*]] = "tf.AddV2"(%[[conv]], %[[cst_0]])
Dgpu_fusion.mlir19 %add = "tf.AddV2"(%arg0, %y#0) : (tensor<8x8x8x8xf32>, tensor<8x8x8x8xf32>) -> tensor<8x8x8x8xf32>
26 // Since the tf.AddV2 op has two uses, we have a _FusedBatchNormEx without the
32 %add = "tf.AddV2"(%arg0, %y#0) : (tensor<8x8x8x8xf32>, tensor<8x8x8x8xf32>) -> tensor<8x8x8x8xf32>
Dresource_op_lifting.mlir177 // CHECK-NEXT: %[[ADD0:.*]] = "tf.AddV2"(%[[BARG1]], %[[BARG1]])
178 %add0 = "tf.AddV2"(%read0, %read0) : (tensor<f32>, tensor<f32>) -> tensor<f32>
181 // CHECK-NEXT: %[[ADD1:.*]] = "tf.AddV2"(%[[ADD0]], %[[ADD0]])
182 %add1 = "tf.AddV2"(%read1, %read1) : (tensor<f32>, tensor<f32>) -> tensor<f32>
186 // CHECK-NEXT: %[[ADD2:.*]] = "tf.AddV2"(%[[BARG0]], %[[DELTA]])
187 %add2 = "tf.AddV2"(%arg0, %constant) : (tensor<i32>, tensor<i32>) -> tensor<i32>
403 // CHECK-NEXT: %[[ADD:.*]] = "tf.AddV2"(%[[CASE]]#1, %[[CASE]]#0)
405 %5 = "tf.AddV2"(%4, %3#1) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
461 // CHECK-NEXT: %[[ADD:.*]] = "tf.AddV2"(%[[IF]]#1, %[[IF]]#0)
463 %5 = "tf.AddV2"(%4, %3#1) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32>
[all …]
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DAddV2.pbtxt2 name: "AddV2"
37 name: "AddV2"
73 name: "AddV2"
109 name: "AddV2"
146 name: "AddV2"
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf-while.mlir18 …%0 = "tf.AddV2"(%arg2, %cst_2) {T = i32, device = ""} : (tensor<*xi32>, tensor<i32>) -> tensor<*xi…
25 …%7 = "tf.AddV2"(%arg4, %6) {T = f32, device = ""} : (tensor<*xf32>, tensor<?x?xf32>) -> tensor<*xf…
26 …%8 = "tf.AddV2"(%arg0, %cst_2) {T = i32, device = ""} : (tensor<*xi32>, tensor<i32>) -> tensor<*xi…
70 // CANON: [[VAL_14:%.*]] = "tf.AddV2"([[VAL_12]], [[VAL_4]])
71 // CANON: [[VAL_15:%.*]] = "tf.AddV2"([[VAL_13]], [[VAL_5]])
72 // CANON: [[VAL_16:%.*]] = "tf.AddV2"([[VAL_11]], [[VAL_4]])
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DAddV2.pbtxt2 name: "AddV2"
37 name: "AddV2"
73 name: "AddV2"
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/tf_to_kernel/
Dadd_v2.mlir3 func @AddV2(%arg0: tensor<*xf32>, %arg1: tensor<*xf32>)
5 %0 = "tf.AddV2"(%arg0, %arg1) {T = f32, device = ""}
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Doptimize.td72 // Mul AddV2
74 // Here, root of the result is AddV2 instead of BiasAdd because the value may
77 // to AddV2 op.
93 // AddV2 value Mul Mul
95 // Mul AddV2
/external/tensorflow/tensorflow/compiler/mlir/tfjs/tests/
Doptimize.mlir11 %4 = "tf.AddV2"(%0, %3) : (tensor<2x3xf32>, tensor<2x3xf32>) -> tensor<2x3xf32>
25 %4 = "tf.AddV2"(%0, %3) : (tensor<2x3xf32>, tensor<2x3xf32>) -> tensor<2x3xf32>
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dflex_exclusively.mlir22 // CHECK-NEXT: name: "tf.AddV2",
53 %0 = "tf.AddV2"(%arg0, %arg0) : (tensor<3x2xf32>, tensor<3x2xf32>) -> tensor<3x2xf32>
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dcase.mlir19 …%outputs_0, %control_1 = tf_executor.island wraps "tf.AddV2"(%arg0, %outputs) {device = ""} : (ten…
28 …%outputs_0, %control_1 = tf_executor.island wraps "tf.AddV2"(%arg0, %outputs) {device = ""} : (ten…
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_AddV2.pbtxt2 graph_op_name: "AddV2"
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_AddV2.pbtxt2 graph_op_name: "AddV2"
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-full-conversion.mlir29 %0 = "tf.AddV2"(%arg0, %arg0) : (tensor<2xi32>, tensor<2xi32>) -> tensor<2xi32>
41 %2 = "tf.AddV2"(%1, %1) : (tensor<2xi32>, tensor<2xi32>) -> tensor<2xi32>

123