| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 63 Eigen::IndexPair<Eigen::DenseIndex> ContractionDims(bool adj_x, bool adj_y) { in ContractionDims() argument
64 return Eigen::IndexPair<Eigen::DenseIndex>(adj_x ? 0 : 1, adj_y ? 1 : 0); in ContractionDims()
78 const Tensor in_y, bool adj_x, bool adj_y, bool trans_x, in Run()
91 contract_pairs[0] = ContractionDims(adj_x || trans_x, adj_y || trans_y); in Run()
104 if (adj_x != adj_y) { in Run()
123 const Tensor& in_y, bool adj_x, bool adj_y, bool trans_x,
129 contract_pairs[0] = ContractionDims(adj_x || trans_x, adj_y || trans_y);
179 bool adj_y, bool trans_x, bool trans_y,
193 if (!adj_y && !trans_y) {
195 } else if (adj_y) {
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | batchmatmul_to_einsum.mlir | 6 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<1x2x3xf32>, tensor<3x…
13 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<2x2x4xf32>, tensor<2x…
20 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<1x2x?xf32>, tensor<?x…
27 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<1x2x?xf32>, tensor<3x…
35 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = true} : (tensor<1x2x3xf32>, tensor<4x3…
42 …%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = true, adj_y = false} : (tensor<1x3x2xf32>, tensor<3…
|
| D | einsum.mlir | 7 …// CHECK: "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<3x4x5xf32>, te…
14 …// CHECK: %[[v0:.*]] = "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<7…
22 …// CHECK: "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<3x4x5xf32>, te…
29 …// CHECK: "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<3x4x5x6x7xf32>…
41 …// CHECK: %[[v2:.*]] = "tf.BatchMatMulV2"(%[[v0]], %[[v1]]) {adj_x = false, adj_y = false} : (tens…
54 …// CHECK: %[[v2:.*]] = "tf.BatchMatMulV2"(%[[v0]], %[[v1]]) {adj_x = false, adj_y = false} : (tens…
66 …// CHECK: "tf.BatchMatMulV2"(%[[v0]], %[[v1]]) {adj_x = false, adj_y = false} : (tensor<2x7x5x3xf3…
77 …// CHECK: "tf.BatchMatMulV2"(%[[v0]], %[[v1]]) {adj_x = false, adj_y = false} : (tensor<2x5x21xf32…
87 …// CHECK: %[[v1:.*]] = "tf.BatchMatMulV2"(%arg0, %[[v0]]) {adj_x = false, adj_y = false} : (tensor…
99 …// CHECK: %[[v1:.*]] = "tf.BatchMatMulV2"(%[[v0]], %arg1) {adj_x = false, adj_y = false} : (tensor…
[all …]
|
| D | fold-broadcast.mlir | 80 …%1 = "tf.BatchMatMulV2"(%arg0, %0) {adj_x = false, adj_y = false} : (tensor<17x17x17xf32>, tensor<…
82 …// CHECK: %[[V0:.*]] = "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<1…
90 …%1 = "tf.BatchMatMulV2"(%0, %arg1) {adj_x = false, adj_y = false} : (tensor<17x17x17xf32>, tensor<…
92 …// CHECK: %[[V0:.*]] = "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<1…
100 …%1 = "tf.BatchMatMulV2"(%0, %arg1) {adj_x = false, adj_y = false} : (tensor<17x17x17xf32>, tensor<…
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BatchMatMul.pbtxt | 37 name: "adj_y"
81 name: "adj_y"
125 name: "adj_y"
170 name: "adj_y"
|
| D | BatchMatMulV2.pbtxt | 39 name: "adj_y"
85 name: "adj_y"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BatchMatMul.pbtxt | 37 name: "adj_y"
81 name: "adj_y"
125 name: "adj_y"
170 name: "adj_y"
|
| D | BatchMatMulV2.pbtxt | 39 name: "adj_y"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf-BatchMatMulV2.mlir | 24 …%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {T = f32, adj_x = false, adj_y = false, device = ""} : (tens…
36 …%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {T = f32, adj_x = false, adj_y = false, device = ""} : (tens…
48 …%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {T = f32, adj_x = false, adj_y = false, device = ""} : (tens…
59 …%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {T = f32, adj_x = false, adj_y = false, device = ""} : (tens…
70 …%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = true, adj_y = true, device = ""} : (tensor<2x5xf32>…
87 …%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = true, adj_y = true, device = ""} : (tensor<2x5xcomp…
|
| D | legalize-tf-include-tf2xla-fallback.mlir | 55 …%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {T = f32, adj_x = false, adj_y = false, device = ""} : (tens…
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchMatMul.pbtxt | 28 name: "adj_y"
40 the `adj_x` or `adj_y` flag to `True`, which are by default `False`.
48 c_o = r_y if adj_y else c_y
|
| D | api_def_BatchMatMulV2.pbtxt | 28 name: "adj_y"
40 the `adj_x` or `adj_y` flag to `True`, which are by default `False`.
48 c_o = r_y if adj_y else c_y
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 102 bool adj_x, bool adj_y, int output_rank, in ResizeOutputTensor() argument
117 int rhs_cols_index = adj_y ? output_rank - 2 : output_rank - 1; in ResizeOutputTensor()
300 bool adj_y = op_context.params->adj_y; in Prepare() local
377 int accum_dim_rhs = adj_y ? extended_rhs_shape.Dims(output_rank - 1) in Prepare()
383 adj_y, output_rank, output); in Prepare()
666 bool adj_y = op_context.params->adj_y; in Eval() local
669 const TfLiteTensor* rhs_tensor = adj_y ? rhs : GetTempRhs(context, node, rhs); in Eval()
671 if (!adj_y) { in Eval()
683 adj_y ? orig_rhs_shape : SwapRowColumnDims(orig_rhs_shape); in Eval()
|
| D | batch_matmul_test.cc | 46 bool adj_x = false, bool adj_y = false) { in BatchMatMulOpModel() argument
52 CreateBatchMatMulOptions(builder_, adj_x, adj_y).Union()); in BatchMatMulOpModel()
287 bool adj_x = false, bool adj_y = false) { in ConstRHSBatchMatMulOpModel() argument
295 CreateBatchMatMulOptions(builder_, adj_x, adj_y).Union(), in ConstRHSBatchMatMulOpModel()
714 bool adj_x = false, bool adj_y = false) argument
734 CreateBatchMatMulOptions(builder_, adj_x, adj_y).Union());
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | canonicalize.td | 65 $adj_x, $adj_y),
66 (TF_BatchMatMulV2Op $x, $y, $adj_x, $adj_y)>;
68 def BatchMatMulToMatMul : Pat<(TF_BatchMatMulOp $x, $y, $adj_x, $adj_y),
69 (TF_MatMulOp $x, $y, $adj_x, $adj_y),
76 def BatchMatMulV2ToMatMul : Pat<(TF_BatchMatMulV2Op $x, $y, $adj_x, $adj_y),
77 (TF_MatMulOp $x, $y, $adj_x, $adj_y),
|
| D | fold_broadcast.cc | 105 matmul_op.adj_y() ? shape_y.back() : *(shape_y.rbegin() + 1); in RewriteBatchMatMulV2Op()
107 !matmul_op.adj_y() ? shape_y.back() : *(shape_y.rbegin() + 1); in RewriteBatchMatMulV2Op()
|
| D | batchmatmul_to_einsum.cc | 74 if (op.adj_y()) std::swap(equation[6 + 3], equation[6 + 4]); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | prepare-composite-functions-tf.mlir | 13 …%8 = "tf.BatchMatMulV2"(%7, %arg0) {adj_x = false, adj_y = false} : (tensor<*xf32>, tensor<*xf32>)…
18 …%0 = "tf.BatchMatMulV2"(%arg3, %arg1) {adj_x = false, adj_y = false} : (tensor<1x3xf32>, tensor<3x…
26 …%0 = "tf.BatchMatMulV2"(%arg3, %arg1) {adj_x = false, adj_y = false} : (tensor<1x3xf32>, tensor<3x…
160 …%0 = "tf.BatchMatMulV2"(%arg3, %arg1) {adj_x = false, adj_y = false} : (tensor<*xf32>, tensor<*xf3…
169 …%0 = "tf.BatchMatMulV2"(%arg3, %arg1) {adj_x = false, adj_y = false} : (tensor<*xf32>, tensor<*xf3…
182 …%0 = "tf.BatchMatMulV2"(%arg0, %arg3) {adj_x = false, adj_y = false} : (tensor<?x8x8xf32>, tensor<…
184 …%2 = "tf.BatchMatMulV2"(%1, %arg4) {adj_x = false, adj_y = true} : (tensor<?x8x40xf32>, tensor<10x…
223 …%0 = "tf.BatchMatMulV2"(%arg0, %arg3) {adj_x = false, adj_y = false} : (tensor<8x8x8xf32>, tensor<…
225 …%2 = "tf.BatchMatMulV2"(%1, %arg4) {adj_x = false, adj_y = true} : (tensor<8x8x40xf32>, tensor<10x…
265 …%0 = "tf.BatchMatMulV2"(%arg0, %arg3) {adj_x = false, adj_y = false} : (tensor<?x8x8xf32>, tensor<…
[all …]
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 226 bool adj_y; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_patterns.td | 284 (TF_BatchMatMulV2Op $lhs, $rhs, $adj_x, $adj_y),
285 (TFL_BatchMatMulOp $lhs, $rhs, $adj_x, $adj_y)>;
287 (TF_BatchMatMulOp $lhs, $rhs, $adj_x, $adj_y),
288 (TFL_BatchMatMulOp $lhs, $rhs, $adj_x, $adj_y)>;
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/ |
| D | unroll_batch_matmul.pbtxt | 70 key: "adj_y"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops_test.cc | 411 auto set_adj = [&op](bool adj_x, bool adj_y) { in TEST() argument
416 .Attr("adj_y", adj_y) in TEST()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.h | 85 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y);
|
| D | testlib.cc | 171 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y) { in BatchMatmul() argument
177 .Attr("adj_y", adj_y) in BatchMatmul()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 317 bool adj_y; in BatchMatMulV2Shape() local
319 TF_RETURN_IF_ERROR(c->GetAttr("adj_y", &adj_y)); in BatchMatMulV2Shape()
321 DimensionHandle output_cols = c->Dim(b_shape, adj_y ? -2 : -1); in BatchMatMulV2Shape()
326 c->Dim(b_shape, adj_y ? -1 : -2), &inner_merged)); in BatchMatMulV2Shape()
354 bool adj_y; in BatchMatMulShape() local
356 TF_RETURN_IF_ERROR(c->GetAttr("adj_y", &adj_y)); in BatchMatMulShape()
358 DimensionHandle output_cols = c->Dim(b_shape, adj_y ? -2 : -1); in BatchMatMulShape()
371 c->Dim(b_shape, adj_y ? -1 : -2), &unused)); in BatchMatMulShape()
|