Home
last modified time | relevance | path

Searched refs:adj_y (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dbatch_matmul_op_impl.h66 Eigen::IndexPair<Eigen::DenseIndex> ContractionDims(bool adj_x, bool adj_y) { in ContractionDims() argument
67 return Eigen::IndexPair<Eigen::DenseIndex>(adj_x ? 0 : 1, adj_y ? 1 : 0); in ContractionDims()
81 const Tensor in_y, bool adj_x, bool adj_y, in Run()
93 contract_pairs[0] = ContractionDims(adj_x, adj_y); in Run()
105 if (adj_x != adj_y) { in Run()
124 const Tensor& in_y, bool adj_x, bool adj_y,
130 contract_pairs[0] = ContractionDims(adj_x, adj_y);
172 bool adj_y, const MatMulBCast& bcast, Tensor* out, int start,
184 if (!adj_y) {
190 if (!adj_y) {
[all …]
Dbatch_matmul_op_test.cc41 Node* BatchMatmulV2(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y) { in BatchMatmulV2() argument
47 .Attr("adj_y", adj_y) in BatchMatmulV2()
Deinsum_op_impl.h588 bool adj_y = !swap_free_and_contract[1]; in ContractOperands() local
591 if (adj_y) TF_RETURN_IF_ERROR(Conjugate<Device, T>(ctx, &rhs)); in ContractOperands()
603 LaunchBatchMatMul<Device, T>::Launch(ctx, lhs, rhs, adj_x, adj_y, bcast, in ContractOperands()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DBatchMatMul.pbtxt37 name: "adj_y"
81 name: "adj_y"
125 name: "adj_y"
170 name: "adj_y"
DBatchMatMulV2.pbtxt39 name: "adj_y"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BatchMatMul.pbtxt28 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
Dapi_def_BatchMatMulV2.pbtxt28 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/compiler/mlir/tensorflow/transforms/
Dcanonicalize.td51 def BatchMatMulToMatMul : Pat<(TF_BatchMatMulOp $x, $y, $adj_x, $adj_y),
52 (TF_MatMulOp $x, $y, $adj_x, $adj_y),
59 def BatchMatMulV2ToMatMul : Pat<(TF_BatchMatMulV2Op $x, $y, $adj_x, $adj_y),
60 (TF_MatMulOp $x, $y, $adj_x, $adj_y),
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dprepare-composite-functions-tf.mlir13 …%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…
158 …%0 = "tf.BatchMatMulV2"(%arg0, %arg3) {adj_x = false, adj_y = false} : (tensor<?x8x8xf32>, tensor<…
160 …%2 = "tf.BatchMatMulV2"(%1, %arg4) {adj_x = false, adj_y = true} : (tensor<?x8x40xf32>, tensor<10x…
/external/tensorflow/tensorflow/core/ops/
Dmath_ops_test.cc411 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/framework/
Dcommon_shape_fns.cc426 bool adj_y; in BatchMatMulV2Shape() local
428 TF_RETURN_IF_ERROR(c->GetAttr("adj_y", &adj_y)); in BatchMatMulV2Shape()
430 DimensionHandle output_cols = c->Dim(b_shape, adj_y ? -2 : -1); in BatchMatMulV2Shape()
435 c->Dim(b_shape, adj_y ? -1 : -2), &inner_merged)); in BatchMatMulV2Shape()
463 bool adj_y; in BatchMatMulShape() local
465 TF_RETURN_IF_ERROR(c->GetAttr("adj_y", &adj_y)); in BatchMatMulShape()
467 DimensionHandle output_cols = c->Dim(b_shape, adj_y ? -2 : -1); in BatchMatMulShape()
480 c->Dim(b_shape, adj_y ? -1 : -2), &unused)); in BatchMatMulShape()
Dcommon_shape_fns_test.cc341 auto set_adj = [&op](bool adj_x, bool adj_y) { in TEST() argument
346 .Attr("adj_y", adj_y) in TEST()
/external/tensorflow/tensorflow/core/graph/
Dtestlib.h85 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y);
Dtestlib.cc171 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/python/ops/
Dmath_grad.py1781 adj_y = op.get_attr("adj_y")
1784 if not adj_y:
1791 if not adj_y:
1807 adj_y = op.get_attr("adj_y")
1810 if not adj_y:
1817 if not adj_y:
Dmath_ops.py2881 a, b, adj_x=adjoint_a, adj_y=adjoint_b, name=name)
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc160 if (batch_op->adj_y) { in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dunroll_batch_matmul.cc241 if (op.adj_y()) { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dcanonicalize.mlir21 …%0 = "tf.BatchMatMul"(%arg0, %arg1) {adj_x = false, adj_y = false} : (tensor<2x3xf32>, tensor<3x2x…
30 …%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = true, adj_y = false} : (tensor<4x3xf32>, tensor<4x5…
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2291 adj_y = pfor_input.get_attr("adj_y")
2295 output = math_ops.matmul(x, y, adjoint_a=adj_x, adjoint_b=adj_y)
2306 adj_y = pfor_input.get_attr("adj_y")
2308 output = math_ops.matmul(x, y, adjoint_a=adj_x, adjoint_b=adj_y)
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h1031 bool adj_y = false;
Dimport_tensorflow.cc1316 batch_matmul->adj_y = GetBoolAttr(node, "adj_y"); in ConvertBatchMatMulOperator()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1211 bool adj_y = random_bool(generator()); in TEST_F() local
1215 if (adj_y) { in TEST_F()
1224 .Attr("adj_y", adj_y)); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td355 the `adj_x` or `adj_y` flag to `True`, which are by default `False`.
363 c_o = r_y if adj_y else c_y
375 DefaultValuedAttr<BoolAttr, "false">:$adj_y
396 the `adj_x` or `adj_y` flag to `True`, which are by default `False`.
404 c_o = r_y if adj_y else c_y
420 DefaultValuedAttr<BoolAttr, "false">:$adj_y
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt353 …argspec: "args=[\'x\', \'y\', \'adj_x\', \'adj_y\', \'name\'], varargs=None, keywords=None, defaul…
357 …argspec: "args=[\'x\', \'y\', \'adj_x\', \'adj_y\', \'name\'], varargs=None, keywords=None, defaul…

12