| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_matmul_op_impl.h | 62 Eigen::IndexPair<Eigen::DenseIndex> ContractionDims(bool adj_x, bool adj_y) { in ContractionDims() argument
63 return Eigen::IndexPair<Eigen::DenseIndex>(adj_x ? 0 : 1, adj_y ? 1 : 0); in ContractionDims()
77 const Tensor in_y, bool adj_x, bool adj_y, Tensor* out, in Run()
89 contract_pairs[0] = ContractionDims(adj_x, adj_y); in Run()
94 if (adj_x != adj_y) { in Run()
113 const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out,
119 contract_pairs[0] = ContractionDims(adj_x, adj_y);
133 static void Multiply(bool adj_x, bool adj_y, Tx x, Ty y, Tz z) {
134 if (!adj_x) {
172 static void Run(const Tensor& in_x, const Tensor& in_y, bool adj_x,
[all …]
|
| /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
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | math_utils_test.py | 101 adj_x=True).eval())
109 adj_x=True, adj_y=True).eval())
124 adj_x=True).eval())
132 adj_x=True, adj_y=True).eval())
|
| D | math_utils.py | 270 def batch_times_matrix(batch, matrix, adj_x=False, adj_y=False): argument
291 if adj_x:
314 def matrix_times_batch(matrix, batch, adj_x=False, adj_y=False): argument
318 batch=batch, matrix=matrix, adj_x=not adj_y, adj_y=not adj_x),
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops_test.cc | 392 auto set_adj = [&op](bool adj_x, bool adj_y) { in TEST() argument
396 .Attr("adj_x", adj_x) in TEST()
|
| D | math_ops.cc | 134 bool adj_x; in __anonb22bfa860202() local
136 TF_RETURN_IF_ERROR(c->GetAttr("adj_x", &adj_x)); in __anonb22bfa860202()
138 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in __anonb22bfa860202()
151 TF_RETURN_IF_ERROR(c->Merge(c->Dim(a_shape, adj_x ? -2 : -1), in __anonb22bfa860202()
|
| D | ops.pbtxt | 3337 name: "adj_x"
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/ |
| D | kalman_filter.py | 291 kalman_gain_transposed, observation_noise, adj_x=True),
|
| /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 | 234 if (batch_op->adj_x) { in Run()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_grad.py | 1249 adj_x = op.get_attr("adj_x")
1252 if not adj_x:
|
| D | math_ops.py | 2542 a, b, adj_x=adjoint_a, adj_y=adjoint_b, name=name)
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 983 bool adj_x = false;
|
| D | import_tensorflow.cc | 1098 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/python/ops/parallel_for/ |
| D | pfor.py | 1976 adj_x = pfor_input.get_attr("adj_x")
1981 output = math_ops.matmul(x, y, adjoint_a=adj_x, adjoint_b=adj_y)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 1559 adj_x = None if axes[0] == ndim(x) - 1 else True
1561 out = math_ops.matmul(x, y, adjoint_a=adj_x, adjoint_b=adj_y)
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 4636 name: "adj_x"
4677 name: "adj_x"
4719 name: "adj_x"
4762 name: "adj_x"
|
| D | ops_history.v2.pbtxt | 9250 name: "adj_x"
9294 name: "adj_x"
9338 name: "adj_x"
9383 name: "adj_x"
|
| D | ops_history.v1.pbtxt | 9250 name: "adj_x"
9294 name: "adj_x"
9338 name: "adj_x"
9383 name: "adj_x"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 317 …argspec: "args=[\'x\', \'y\', \'adj_x\', \'adj_y\', \'name\'], varargs=None, keywords=None, defaul…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 317 …argspec: "args=[\'x\', \'y\', \'adj_x\', \'adj_y\', \'name\'], varargs=None, keywords=None, defaul…
|