Home
last modified time | relevance | path

Searched refs:dot_general (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlower-general-dot.mlir8 …%0 = "mhlo.dot_general"(%arg0, %arg1) {dot_dimension_numbers = {lhs_batching_dimensions = dense<[]…
23 …%0 = "mhlo.dot_general"(%arg0, %arg1) {dot_dimension_numbers = {lhs_batching_dimensions = dense<[]…
31 // CHECK-NEXT: "mhlo.dot_general"(%arg0, %arg1)
32 …%0 = "mhlo.dot_general"(%arg0, %arg1) {dot_dimension_numbers = {lhs_batching_dimensions = dense<[0…
Dhlo-legalize-to-linalg.mlir1005 func @dot_general(%arg0: tensor<?x?x3xf32>,
1007 %0 = "mhlo.dot_general"(%arg0, %arg1) {
1018 // CHECK: func @dot_general(%[[ARG0:.*]]: tensor<?x?x3xf32>, %[[ARG1:.*]]: tensor<?x3x?xf32>)
1035 %0 = "mhlo.dot_general"(%arg0, %arg1) {
Dops.mlir1214 func @dot_general(%arg0: tensor<?x?x?xf32>, %arg1: tensor<?x?x?xf32>) {
1216 %0 = "mhlo.dot_general"(%arg0, %arg1) { dot_dimension_numbers = {
1227 func @dot_general(%arg0: tensor<?x?x?xf32>, %arg1: tensor<?x?x?xf32>) {
1229 %0 = "mhlo.dot_general"(%arg0, %arg1) { dot_dimension_numbers = {
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-BatchMatMulV2.mlir20 // CHECK: [[RESULT:%.*]] = "mhlo.dot_general"([[LHSBCAST]], [[RHS]]) {dot_dimension_numbe…
31 // CHECK: "mhlo.dot_general"({{.*}}, {{.*}}) {dot_dimension_numbers = {
43 // CHECK: "mhlo.dot_general"({{.*}}, {{.*}}) {dot_dimension_numbers = {
54 // CHECK: "mhlo.dot_general"({{.*}}, {{.*}}) {dot_dimension_numbers = {
65 // CHECK: "mhlo.dot_general"({{.*}}, {{.*}}) {dot_dimension_numbers = {
Dlegalize-tf-include-tf2xla-fallback.mlir41 // The fallback pattern uses dot_general without broadcast on operands and then
49 // NO_FALLBACK: mhlo.dot_general
52 // SUPPORTED_FALLBACK_DEVICE: mhlo.dot_general
/external/tensorflow/tensorflow/compiler/tests/
Dxla_ops_test.py182 return xla.dot_general(
558 xla.dot_general(a, b, dim_nums)
571 xla.dot_general(a, b, dim_nums)
585 xla.dot_general(a, b, dim_nums)
600 xla.dot_general(a, b, dim_nums)
612 c = xla.dot_general(a, b, dim_nums)
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py297 def dot_general(lhs, rhs, dimension_numbers, precision_config=None, name=None): function
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dimport.hlotxt331 …// CHECK-NEXT: [[R0:%.+]] = "mhlo.dot_general"([[ARG0]], [[ARG1]]) {dot_dimension_numbers = {lhs_…
334 …// CHECK-NEXT: [[R1:%.+]] = "mhlo.dot_general"([[ARG0]], [[ARG1]]) {dot_dimension_numbers = {lhs_…
337 …// CHECK-NEXT: [[R2:%.+]] = "mhlo.dot_general"([[ARG0]], [[ARG1]]) {dot_dimension_numbers = {lhs_…
341 …// CHECK-NEXT: "mhlo.dot_general"([[ARG0]], [[ARG1]]) {dot_dimension_numbers = {lhs_batching_dime…
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.td952 def HLO_DotGeneralOp: HLO_Op<"dot_general", [NoSideEffect]>,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlegalize_hlo.mlir1596 …%0 = "mhlo.dot_general"(%arg0, %arg1) {dot_dimension_numbers = {lhs_batching_dimensions = dense<0>…
1611 …%0 = "mhlo.dot_general"(%arg0, %arg1) {dot_dimension_numbers = {lhs_batching_dimensions = dense<> …