| /external/tensorflow/tensorflow/core/kernels/ |
| batch_matmul_op_impl.h | 48 Eigen::IndexPair<Eigen::DenseIndex> ContractionDims(bool adj_x, bool adj_y) {
50 if (!adj_y) {
56 if (!adj_y) {
75 const Tensor in_y, bool adj_x, bool adj_y, Tensor* out,
87 contract_pairs[0] = ContractionDims(adj_x, adj_y);
92 if (adj_x != adj_y) {
111 const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out,
117 contract_pairs[0] = ContractionDims(adj_x, adj_y);
131 static void Multiply(bool adj_x, bool adj_y, Tx x, Ty y, Tz z) {
133 if (!adj_y) {
[all...] |
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| math_utils_test.py | 105 adj_y=True).eval())
109 adj_x=True, adj_y=True).eval())
128 adj_y=True).eval())
132 adj_x=True, adj_y=True).eval())
|
| math_utils.py | 242 def batch_times_matrix(batch, matrix, adj_x=False, adj_y=False):
248 adjoint_a=adj_x, adjoint_b=adj_y)
255 adj_y: If true, transpose "matrix" before multiplying.
274 if adj_y:
283 math_ops.matmul(batch_reshaped, matrix, adjoint_b=adj_y), result_shape)
286 def matrix_times_batch(matrix, batch, adj_x=False, adj_y=False):
290 batch=batch, matrix=matrix, adj_x=not adj_y, adj_y=not adj_x),
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| math_ops_test.cc | 386 auto set_adj = [&op](bool adj_x, bool adj_y) {
391 .Attr("adj_y", adj_y)
|
| math_ops.cc | 70 .Attr("adj_y: bool = false")
79 bool adj_y;
81 TF_RETURN_IF_ERROR(c->GetAttr("adj_y", &adj_y));
83 DimensionHandle output_cols = c->Dim(b_shape, adj_y ? -2 : -1);
96 c->Dim(b_shape, adj_y ? -1 : -2), &unused));
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| testlib.h | 81 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y);
|
| testlib.cc | 183 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y) {
189 .Attr("adj_y", adj_y)
|
| /external/tensorflow/tensorflow/python/ops/ |
| math_grad.py | [all...] |
| math_ops.py | [all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | [all...] |
| /external/tensorflow/tensorflow/python/keras/_impl/keras/ |
| backend.py | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
