| /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);
178 static void Run(const Tensor& in_x, const Tensor& in_y, bool adj_x,
192 if (!adj_x && !trans_x) {
200 } else if (adj_x) {
[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 | 30 name: "adj_x"
74 name: "adj_x"
118 name: "adj_x"
163 name: "adj_x"
|
| D | BatchMatMulV2.pbtxt | 32 name: "adj_x"
78 name: "adj_x"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BatchMatMul.pbtxt | 30 name: "adj_x"
74 name: "adj_x"
118 name: "adj_x"
163 name: "adj_x"
|
| D | BatchMatMulV2.pbtxt | 32 name: "adj_x"
|
| /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 | 22 name: "adj_x"
40 the `adj_x` or `adj_y` flag to `True`, which are by default `False`.
47 r_o = c_x if adj_x else r_x
|
| D | api_def_BatchMatMulV2.pbtxt | 22 name: "adj_x"
40 the `adj_x` or `adj_y` flag to `True`, which are by default `False`.
47 r_o = c_x if adj_x else r_x
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 102 bool adj_x, bool adj_y, int output_rank, in ResizeOutputTensor() argument
116 int lhs_rows_index = adj_x ? output_rank - 1 : output_rank - 2; in ResizeOutputTensor()
147 const int batch_size = op_context->params->adj_x in InitializeTemporaries()
299 bool adj_x = op_context.params->adj_x; in Prepare() local
375 int accum_dim_lhs = adj_x ? extended_lhs_shape.Dims(output_rank - 2) in Prepare()
382 ResizeOutputTensor(context, extended_lhs_shape, extended_rhs_shape, adj_x, in Prepare()
667 bool adj_x = op_context.params->adj_x; in Eval() local
670 const TfLiteTensor* lhs_tensor = adj_x ? GetTempLhs(context, node, lhs) : lhs; in Eval()
679 if (adj_x) { in Eval()
685 adj_x ? orig_lhs_shape : SwapRowColumnDims(orig_lhs_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 | 100 matmul_op.adj_x() ? shape_x.back() : *(shape_x.rbegin() + 1); in RewriteBatchMatMulV2Op()
102 !matmul_op.adj_x() ? shape_x.back() : *(shape_x.rbegin() + 1); in RewriteBatchMatMulV2Op()
|
| D | batchmatmul_to_einsum.cc | 73 if (op.adj_x()) std::swap(equation[3], equation[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 | 225 bool adj_x; 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 | 64 key: "adj_x"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops_test.cc | 411 auto set_adj = [&op](bool adj_x, bool adj_y) { in TEST() argument
415 .Attr("adj_x", adj_x) 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
176 .Attr("adj_x", adj_x) in BatchMatmul()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 316 bool adj_x; in BatchMatMulV2Shape() local
318 TF_RETURN_IF_ERROR(c->GetAttr("adj_x", &adj_x)); in BatchMatMulV2Shape()
320 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in BatchMatMulV2Shape()
325 TF_RETURN_IF_ERROR(c->Merge(c->Dim(a_shape, adj_x ? -2 : -1), in BatchMatMulV2Shape()
353 bool adj_x; in BatchMatMulShape() local
355 TF_RETURN_IF_ERROR(c->GetAttr("adj_x", &adj_x)); in BatchMatMulShape()
357 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in BatchMatMulShape()
370 TF_RETURN_IF_ERROR(c->Merge(c->Dim(a_shape, adj_x ? -2 : -1), in BatchMatMulShape()
|