Searched refs:adj_y (Results 1 - 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dbatch_matmul_op_impl.h48 Eigen::IndexPair<Eigen::DenseIndex> ContractionDims(bool adj_x, bool adj_y) { argument
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) { argument
133 if (!adj_y) {
74 Run(const OpKernelContext* context, const Tensor& in_x, const Tensor in_y, bool adj_x, bool adj_y, Tensor* out, int start, int limit) argument
110 Run(const OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out, int start, int limit) argument
170 Run(const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out, int start, int limit) argument
201 Launch(OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out) argument
290 Launch(OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out) argument
410 Run(const OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out, int start, int limit) argument
430 Launch(OpKernelContext* context, const Tensor& in_x, const Tensor& in_y, bool adj_x, bool adj_y, Tensor* out) argument
[all...]
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
H A Dmath_utils_test.py105 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())
H A Dmath_utils.py242 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),
/external/tensorflow/tensorflow/core/ops/
H A Dmath_ops_test.cc386 auto set_adj = [&op](bool adj_x, bool adj_y) {
391 .Attr("adj_y", adj_y)
H A Dmath_ops.cc70 .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));
/external/tensorflow/tensorflow/core/graph/
H A Dtestlib.h81 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y);
H A Dtestlib.cc183 Node* BatchMatmul(Graph* g, Node* in0, Node* in1, bool adj_x, bool adj_y) { argument
189 .Attr("adj_y", adj_y)
/external/tensorflow/tensorflow/python/ops/
H A Dmath_grad.py1155 adj_y = op.get_attr("adj_y")
1158 if not adj_y:
1165 if not adj_y:
H A Dmath_ops.py2030 a, b, adj_x=adjoint_a, adj_y=adjoint_b, name=name)
/external/tensorflow/tensorflow/python/keras/_impl/keras/
H A Dbackend.py1404 adj_y = True if axes[1] == ndim(y) - 1 else None
1407 adj_y = None
1408 out = math_ops.matmul(x, y, adjoint_a=adj_x, adjoint_b=adj_y)
/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc1123 bool adj_y = random_bool(generator());
1127 if (adj_y) {
1136 .Attr("adj_y", adj_y));

Completed in 331 milliseconds