/external/tensorflow/tensorflow/core/kernels/ |
D | batch_matmul_op_impl.h | 66 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 …]
|
D | batch_matmul_op_test.cc | 41 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()
|
D | einsum_op_impl.h | 588 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/ |
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/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/compiler/mlir/tensorflow/transforms/ |
D | canonicalize.td | 51 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/ |
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… 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/ |
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/framework/ |
D | common_shape_fns.cc | 426 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()
|
D | common_shape_fns_test.cc | 341 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/ |
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/python/ops/ |
D | math_grad.py | 1781 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:
|
D | math_ops.py | 2881 a, b, adj_x=adjoint_a, adj_y=adjoint_b, name=name)
|
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | unroll_batch_matmul.cc | 160 if (batch_op->adj_y) { in Run()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
D | unroll_batch_matmul.cc | 241 if (op.adj_y()) { in matchAndRewrite()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
D | canonicalize.mlir | 21 …%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/ |
D | pfor.py | 2291 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/ |
D | model.h | 1031 bool adj_y = false;
|
D | import_tensorflow.cc | 1316 batch_matmul->adj_y = GetBoolAttr(node, "adj_y"); in ConvertBatchMatMulOperator()
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 1211 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/ |
D | tf_generated_ops.td | 355 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/ |
D | tensorflow.raw_ops.pbtxt | 353 …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…
|