Searched refs:adj_x (Results 1 - 12 of 12) 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
49 if (!adj_x) {
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
132 if (!adj_x) {
170 static void Run(const Tensor& in_x, const Tensor& in_y, bool adj_x, argument
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
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.py101 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())
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)
253 adj_x: If true, transpose the second two dimensions of "batch" before
263 if adj_x:
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/contrib/timeseries/python/timeseries/state_space_models/
H A Dkalman_filter.py292 kalman_gain_transposed, observation_noise, adj_x=True),
/external/tensorflow/tensorflow/core/ops/
H A Dmath_ops_test.cc386 auto set_adj = [&op](bool adj_x, bool adj_y) {
390 .Attr("adj_x", adj_x)
H A Dmath_ops.cc69 .Attr("adj_x: bool = false")
78 bool adj_x;
80 TF_RETURN_IF_ERROR(c->GetAttr("adj_x", &adj_x));
82 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2);
95 TF_RETURN_IF_ERROR(c->Merge(c->Dim(a_shape, adj_x ? -2 : -1),
/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
188 .Attr("adj_x", adj_x)
/external/tensorflow/tensorflow/python/keras/_impl/keras/
H A Dbackend.py1403 adj_x = None if axes[0] == ndim(x) - 1 else True
1406 adj_x = None
1408 out = math_ops.matmul(x, y, adjoint_a=adj_x, adjoint_b=adj_y)
/external/tensorflow/tensorflow/python/ops/
H A Dmath_grad.py1154 adj_x = op.get_attr("adj_x")
1157 if not adj_x:
H A Dmath_ops.py2030 a, b, adj_x=adjoint_a, adj_y=adjoint_b, name=name)
/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc1122 bool adj_x = random_bool(generator());
1124 if (adj_x) {
1135 .Attr("adj_x", adj_x)

Completed in 2502 milliseconds