Home
last modified time | relevance | path

Searched refs:adj_x (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);
171 static void Run(const Tensor& in_x, const Tensor& in_y, bool adj_x,
183 if (!adj_x) {
208 const Tensor& in_y, bool adj_x, bool 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
46 .Attr("adj_x", adj_x) in BatchMatmulV2()
Deinsum_op_impl.h587 bool adj_x = swap_free_and_contract[0]; in ContractOperands() local
590 if (adj_x) TF_RETURN_IF_ERROR(Conjugate<Device, T>(ctx, &lhs)); 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.pbtxt30 name: "adj_x"
74 name: "adj_x"
118 name: "adj_x"
163 name: "adj_x"
DBatchMatMulV2.pbtxt32 name: "adj_x"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BatchMatMul.pbtxt22 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
Dapi_def_BatchMatMulV2.pbtxt22 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/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
415 .Attr("adj_x", adj_x) in TEST()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc425 bool adj_x; in BatchMatMulV2Shape() local
427 TF_RETURN_IF_ERROR(c->GetAttr("adj_x", &adj_x)); in BatchMatMulV2Shape()
429 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in BatchMatMulV2Shape()
434 TF_RETURN_IF_ERROR(c->Merge(c->Dim(a_shape, adj_x ? -2 : -1), in BatchMatMulV2Shape()
462 bool adj_x; in BatchMatMulShape() local
464 TF_RETURN_IF_ERROR(c->GetAttr("adj_x", &adj_x)); in BatchMatMulShape()
466 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in BatchMatMulShape()
479 TF_RETURN_IF_ERROR(c->Merge(c->Dim(a_shape, adj_x ? -2 : -1), in BatchMatMulShape()
Dcommon_shape_fns_test.cc341 auto set_adj = [&op](bool adj_x, bool adj_y) { in TEST() argument
345 .Attr("adj_x", adj_x) 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
176 .Attr("adj_x", adj_x) in BatchMatmul()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc152 if (batch_op->adj_x) { in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dunroll_batch_matmul.cc233 if (op.adj_x()) { 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/
Dmath_grad.py1780 adj_x = op.get_attr("adj_x")
1783 if not adj_x:
1806 adj_x = op.get_attr("adj_x")
1809 if not adj_x:
Dmath_ops.py2881 a, b, adj_x=adjoint_a, adj_y=adjoint_b, name=name)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2290 adj_x = pfor_input.get_attr("adj_x")
2295 output = math_ops.matmul(x, y, adjoint_a=adj_x, adjoint_b=adj_y)
2305 adj_x = pfor_input.get_attr("adj_x")
2308 output = math_ops.matmul(x, y, adjoint_a=adj_x, adjoint_b=adj_y)
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h1030 bool adj_x = false;
Dimport_tensorflow.cc1313 batch_matmul->adj_x = GetBoolAttr(node, "adj_x"); in ConvertBatchMatMulOperator()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1210 bool adj_x = random_bool(generator()); in TEST_F() local
1212 if (adj_x) { in TEST_F()
1223 .Attr("adj_x", adj_x) 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`.
362 r_o = c_x if adj_x else r_x
374 DefaultValuedAttr<BoolAttr, "false">:$adj_x,
396 the `adj_x` or `adj_y` flag to `True`, which are by default `False`.
403 r_o = c_x if adj_x else r_x
419 DefaultValuedAttr<BoolAttr, "false">:$adj_x,
/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