/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); 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 …]
|
D | batch_matmul_op_test.cc | 41 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()
|
D | einsum_op_impl.h | 587 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/ |
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/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/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 415 .Attr("adj_x", adj_x) in TEST()
|
/external/tensorflow/tensorflow/core/framework/ |
D | common_shape_fns.cc | 425 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()
|
D | common_shape_fns_test.cc | 341 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/ |
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/lite/toco/graph_transformations/ |
D | unroll_batch_matmul.cc | 152 if (batch_op->adj_x) { in Run()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
D | unroll_batch_matmul.cc | 233 if (op.adj_x()) { 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/ |
D | math_grad.py | 1780 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:
|
D | math_ops.py | 2881 a, b, adj_x=adjoint_a, adj_y=adjoint_b, name=name)
|
/external/tensorflow/tensorflow/python/ops/parallel_for/ |
D | pfor.py | 2290 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/ |
D | model.h | 1030 bool adj_x = false;
|
D | import_tensorflow.cc | 1313 batch_matmul->adj_x = GetBoolAttr(node, "adj_x"); in ConvertBatchMatMulOperator()
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 1210 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/ |
D | tf_generated_ops.td | 355 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/ |
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…
|