Home
last modified time | relevance | path

Searched refs:matmul (Results 1 – 25 of 285) sorted by relevance

12345678910>>...12

/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_grad.py105 matmul = math_ops.matmul
115 grad_a_dense = matmul(grad, b, transpose_b=not t_b)
117 grad_a_dense = matmul(b, grad, transpose_a=t_b, transpose_b=True)
121 grad_a_dense = matmul(grad, b, adjoint_b=not adj_b)
123 grad_a_dense = matmul(b, grad, adjoint_a=adj_b, adjoint_b=True)
131 grad_a_dense = matmul(b, grad, transpose_a=True, adjoint_b=True)
134 grad_a_dense = matmul(b, grad, transpose_a=True, transpose_b=True)
143 grad_a_dense = matmul(grad, b, transpose_a=True, transpose_b=not t_b)
145 grad_a_dense = matmul(b, grad, transpose_a=t_b)
150 grad_a_dense = matmul(grad, b, transpose_a=True, adjoint_b=not adj_b)
[all …]
/external/tensorflow/tensorflow/compiler/aot/tests/
Dtfcompile_test.cc248 foo::bar::MatMulComp matmul; in TEST() local
249 matmul.set_thread_pool(&device); in TEST()
250 EXPECT_EQ(matmul.arg0_data(), matmul.arg_data(0)); in TEST()
251 EXPECT_EQ(matmul.arg1_data(), matmul.arg_data(1)); in TEST()
255 matmul.arg0(0, 0) = 1; in TEST()
256 matmul.arg0(0, 1) = 2; in TEST()
257 matmul.arg0(0, 2) = 3; in TEST()
258 matmul.arg0(1, 0) = 4; in TEST()
259 matmul.arg0(1, 1) = 5; in TEST()
260 matmul.arg0(1, 2) = 6; in TEST()
[all …]
/external/tensorflow/tensorflow/python/ops/
Dlinalg_grad.py54 return -math_ops.matmul(
55 ainv, math_ops.matmul(grad, ainv, adjoint_b=True), adjoint_a=True)
471 middle = math_ops.matmul(l, grad, adjoint_a=True)
476 grad_a = math_ops.matmul(
477 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse)
496 qdq = math_ops.matmul(q, dq, adjoint_a=True)
498 rdr = math_ops.matmul(r, dr, adjoint_b=True)
508 grad_a = math_ops.matmul(q, dr + _TriangularSolve(tril, r))
509 grad_b = _TriangularSolve(dq - math_ops.matmul(q, qdq), r)
521 grad_a = -math_ops.matmul(c, grad_b, adjoint_b=True)
[all …]
Dinit_ops.py733 return math_ops.matmul(c, array_ops.matrix_transpose(c))
808 kernel2x2[0, 0] = math_ops.matmul(p1, p2)
809 kernel2x2[0, 1] = math_ops.matmul(p1, (eye - p2))
810 kernel2x2[1, 0] = math_ops.matmul((eye - p1), p2)
811 kernel2x2[1, 1] = math_ops.matmul((eye - p1), (eye - p2))
843 result[i, j] += math_ops.matmul(m1[index1, index2],
875 p[i, j] = math_ops.matmul(orth, p[i, j])
972 result[i] += math_ops.matmul(m1[index], m2[i - index])
1000 p[i] = math_ops.matmul(orth, p[i])
1082 def matmul(p1, p2, p3): function
[all …]
/external/tensorflow/tensorflow/python/eager/
Dtape_test.py38 mm = math_ops.matmul(a, b)
43 math_ops.matmul(dmm, b, transpose_b=True) +
44 math_ops.matmul(array_ops.ones_like(b * dr), b, transpose_b=True),
45 math_ops.matmul(a, dmm, transpose_b=True) +
46 math_ops.matmul(a, array_ops.ones_like(a) * dr, transpose_b=True)
89 mm = math_ops.matmul(a, b)
96 math_ops.matmul(
103 mm = math_ops.matmul(a, b)
110 math_ops.matmul(
117 mm = math_ops.matmul(a, b)
[all …]
Dfunction_test.py144 matmul = def_function.function(math_ops.matmul)
146 sq = matmul(t, t, transpose_a=True)
147 sq2 = matmul(sq, t, transpose_a=True)
483 matmul = def_function.function(math_ops.matmul)
487 return matmul(a, a)
491 self.assertAllEqual(out, math_ops.matmul(t, t).numpy())
494 matmul = def_function.function(math_ops.matmul)
500 return matmul(inputs.a['a'], inputs.b['b'])
505 self.assertAllEqual(out, math_ops.matmul(t, t).numpy())
508 matmul = def_function.function(math_ops.matmul)
[all …]
/external/tensorflow/tensorflow/c/eager/
Dc_api_remote_test.cc87 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecute() local
88 TFE_OpSetDevice(matmul, remote_device_name, status); in TestRemoteExecute()
93 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecute()
113 TFE_DeleteOp(matmul); in TestRemoteExecute()
171 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h1_task2); in TestRemoteExecuteSilentCopies() local
172 TFE_OpSetDevice(matmul, task1_name, status); in TestRemoteExecuteSilentCopies()
177 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecuteSilentCopies()
201 TFE_DeleteOp(matmul); in TestRemoteExecuteSilentCopies()
260 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecuteDeleteContextWithOutstandingRPC() local
261 TFE_OpSetDevice(matmul, remote_device_name, status); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
[all …]
Dc_api_test.cc50 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp() local
51 TFE_DeleteOp(matmul); in BM_InitOp()
72 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_Execute() local
77 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in BM_Execute()
87 TFE_DeleteOp(matmul); in BM_Execute()
387 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopy() local
388 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopy()
392 TFE_Execute(matmul, &retvals[0], &num_retvals, status.get()); in TensorHandleSilentCopy()
394 TFE_DeleteOp(matmul); in TensorHandleSilentCopy()
435 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopyLocal() local
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dlinear_operator_adjoint_test.py128 np.matmul(matrix1, matrix2.T),
130 full_matrix1.matmul(full_matrix2, adjoint_arg=True).to_dense()))
133 np.matmul(matrix1.T, matrix2),
135 full_matrix1.matmul(full_matrix2, adjoint=True).to_dense()))
138 np.matmul(matrix1.T, matrix2.T),
140 full_matrix1.matmul(
152 np.matmul(matrix1, matrix2.conj().T),
154 full_matrix1.matmul(full_matrix2, adjoint_arg=True).to_dense()))
157 np.matmul(matrix1.conj().T, matrix2),
159 full_matrix1.matmul(full_matrix2, adjoint=True).to_dense()))
[all …]
Dlinear_operator_identity_test.py88 y = operator.matmul(x)
152 operator.matmul(x)
163 self.evaluate(operator.matmul(x))
173 operator_matmul = operator.matmul(x)
187 operator_matmul = operator.matmul(x)
209 operator_matmul = operator.matmul(x)
233 operator_matmul = operator.matmul(x)
378 y = operator.matmul(x)
392 operator.matmul(x)
405 self.evaluate(operator.matmul(x))
[all …]
Dlinear_operator_test.py90 return math_ops.matmul(
222 operator_matmul = operator1.matmul(operator1)
239 operator_matmul = operator2.matmul(operator2)
257 operator_matmul = operator1.matmul(operator1)
274 operator_matmul = operator2.matmul(operator2, adjoint_arg=True)
303 self.assertTrue(operator1.matmul(operator2).is_square)
304 self.assertTrue(operator2.matmul(operator1).is_square)
305 self.assertFalse(operator1.matmul(operator3).is_square)
345 op_val = operator.matmul(
347 matmul_val = math_ops.matmul(
[all …]
Dlinear_operator_lower_triangular_test.py96 operator_matmul = operator1.matmul(operator2)
101 math_ops.matmul(
106 operator_matmul = operator2.matmul(operator1)
111 math_ops.matmul(
Dlinear_operator_diag_test.py150 operator_matmul = operator.matmul(x)
151 mat_matmul = math_ops.matmul(mat, x)
165 operator_matmul = operator1.matmul(operator2)
171 operator_matmul = operator2.matmul(operator1)
177 operator_matmul = operator1.matmul(operator3)
183 operator_matmul = operator3.matmul(operator1)
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_low_rank_update.py349 leading_term = l.matmul(x, adjoint=adjoint, adjoint_arg=adjoint_arg)
352 uh_x = math_ops.matmul(u, x, adjoint_a=True, adjoint_b=adjoint_arg)
353 d_uh_x = d.matmul(uh_x, adjoint=adjoint)
354 v_d_uh_x = math_ops.matmul(v, d_uh_x)
357 vh_x = math_ops.matmul(v, x, adjoint_a=True, adjoint_b=adjoint_arg)
358 d_vh_x = d.matmul(vh_x, adjoint=adjoint)
359 u_d_vh_x = math_ops.matmul(u, d_vh_x)
421 vh_linv_rhs = math_ops.matmul(v, linv_rhs, adjoint_a=True)
430 u_capinv_vh_linv_rhs = math_ops.matmul(u, capinv_vh_linv_rhs)
444 vh_linv_u = math_ops.matmul(self.v, linv_u, adjoint_a=True)
Dlinalg_impl.py141 matrix_2 = math_ops.matmul(matrix, matrix)
143 matrix_u = math_ops.matmul(matrix, tmp)
156 matrix_2 = math_ops.matmul(matrix, matrix)
157 matrix_4 = math_ops.matmul(matrix_2, matrix_2)
159 matrix_u = math_ops.matmul(matrix, tmp)
172 matrix_2 = math_ops.matmul(matrix, matrix)
173 matrix_4 = math_ops.matmul(matrix_2, matrix_2)
174 matrix_6 = math_ops.matmul(matrix_4, matrix_2)
176 matrix_u = math_ops.matmul(matrix, tmp)
192 matrix_2 = math_ops.matmul(matrix, matrix)
[all …]
/external/tensorflow/tensorflow/cc/profiler/
Dprofiler_test.cc137 const GraphNodeProto* matmul = ExtractNode(ret, "y"); in TEST_F() local
138 EXPECT_TRUE(matmul); in TEST_F()
139 EXPECT_GT(matmul->exec_micros(), 0); in TEST_F()
141 EXPECT_GT(matmul->accelerator_exec_micros(), 0); in TEST_F()
143 EXPECT_EQ(matmul->accelerator_exec_micros(), 0); in TEST_F()
/external/tensorflow/tensorflow/python/kernel_tests/
Dmatmul_op_test.py92 res = math_ops.matmul(a, b, **kwargs_)
97 res = math_ops.matmul(a, b, **kwargs_)
133 lambda x: math_ops.matmul(x, effective_b_np, **kwargs_),
139 lambda x: math_ops.matmul(effective_a_np, x, **kwargs_),
153 math_ops.matmul(a, b)
164 math_ops.matmul(a, b, transpose_a=True)
174 infix_matmul = operator.matmul
222 d = math_ops.matmul(a, b)
Dcholesky_op_test.py45 middle = math_ops.matmul(l, grad, adjoint_a=True)
49 grad_a = math_ops.matmul(
50 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse)
59 middle = math_ops.matmul(l, grad, adjoint_a=True)
111 verification = math_ops.matmul(chol, chol, adjoint_b=True)
183 matrix1 = math_ops.matmul(matrix1, matrix1, adjoint_a=True)
184 matrix2 = math_ops.matmul(matrix2, matrix2, adjoint_a=True)
263 tensor = math_ops.matmul(
275 tensor = math_ops.matmul(
Deig_op_test.py89 np.matmul(v, v.transpose()), np.eye(32, dtype=np.float32), atol=2e-3)
91 np.matmul(np.matmul(v, np.diag(e)), v.transpose()))
167 a_ev = math_ops.matmul(
168 math_ops.matmul(tf_v, array_ops.matrix_diag(tf_e)),
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dbiasadd_matmul_test.py44 x1 = math_ops.matmul(x, b)
50 x2 = math_ops.matmul(x2, b, transpose_a=True)
55 x3 = math_ops.matmul(x, b, transpose_b=True)
59 x4 = math_ops.matmul(x4, b, transpose_b=True, transpose_a=True)
65 x5 = math_ops.matmul(x, b)
Dbatch_matmul_test.py36 x1 = math_ops.matmul(inp, inp1, name="matmul")
57 x1 = math_ops.matmul(inp, b, name="matmul")
75 x1 = math_ops.matmul(inp, b, name="matmul")
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dautotune_benchmark.py79 math_ops.matmul, num_parallel_calls=dataset_ops.AUTOTUNE)
101 math_ops.matmul, num_parallel_calls=dataset_ops.AUTOTUNE)
122 dataset = dataset.map(math_ops.matmul)
153 return math_ops.matmul(x, y)
157 return a, math_ops.matmul(x, y)
206 math_ops.matmul, num_parallel_calls=dataset_ops.AUTOTUNE)
220 math_ops.matmul, num_parallel_calls=dataset_ops.AUTOTUNE)
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_test.cc104 ops::MatMul matmul = ops::MatMul( in RunMatMulWithBias() local
111 root.WithOpName("with_bias"), matmul, in RunMatMulWithBias()
123 ops::MatMul matmul = ops::MatMul( in RunMatMulWithBiasAndActivation() local
130 root.WithOpName("with_bias"), matmul, in RunMatMulWithBiasAndActivation()
203 Tensor matmul; in VerifyBiasAddTensorsNear() local
206 run_default(lhs, rhs, bias, &matmul); in VerifyBiasAddTensorsNear()
209 ASSERT_EQ(matmul.dtype(), fused_matmul.dtype()); in VerifyBiasAddTensorsNear()
210 ASSERT_EQ(matmul.shape(), fused_matmul.shape()); in VerifyBiasAddTensorsNear()
212 test::ExpectClose(matmul, fused_matmul, /*atol=*/1e-5); in VerifyBiasAddTensorsNear()
/external/tensorflow/tensorflow/examples/speech_commands/
Dmodels.py200 logits = tf.matmul(fingerprint_input, weights) + bias
326 final_fc = tf.matmul(flattened_second_conv, final_fc_weights) + final_fc_bias
427 first_fc = tf.matmul(flattened_first_conv, first_fc_weights) + first_fc_bias
441 second_fc = tf.matmul(second_fc_input, second_fc_weights) + second_fc_bias
455 final_fc = tf.matmul(final_fc_input, final_fc_weights) + final_fc_bias
606 outputs = tf.matmul(activations_time, weights_time)
640 first_fc = tf.matmul(first_dropout, first_fc_weights) + first_fc_bias
654 second_fc = tf.matmul(second_fc_input, second_fc_weights) + second_fc_bias
668 final_fc = tf.matmul(final_fc_input, final_fc_weights) + final_fc_bias
760 tf.matmul(flattened_first_dropout, final_fc_weights) + final_fc_bias)
[all …]
/external/tensorflow/tensorflow/examples/tutorials/mnist/
Dmnist.py64 hidden1 = tf.nn.relu(tf.matmul(images, weights) + biases)
73 hidden2 = tf.nn.relu(tf.matmul(hidden1, weights) + biases)
82 logits = tf.matmul(hidden2, weights) + biases

12345678910>>...12