Searched refs:matmul_result (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_mkldnn_contraction_kernel_test.cc | 133 Matrix matmul_result(m, n); in TEST() local
134 matmul_result.setZero(); in TEST()
135 matmul_result = lhs_mat * rhs_mat; in TEST()
140 Scalar matmul = matmul_result(i % m, i / m); in TEST()
203 Matrix matmul_result(m, n); in TEST() local
204 matmul_result.setZero(); in TEST()
205 matmul_result = lhs_mat * rhs_mat; in TEST()
211 Scalar matmul = matmul_result(i, j); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 203 Tensor* matmul_result = nullptr; in Compute() local
208 &output_transposed, &matmul_result)); in Compute()
212 ctx, batch_size, num_lhs_rows, *sparse_matrix_a, *rhs, matmul_result); in Compute()
216 matmul_result); in Compute()
240 Tensor** matmul_result) { in AllocateOutput() argument
247 *matmul_result = *output; in AllocateOutput()
256 *matmul_result = output_transposed; in AllocateOutput()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad.cc | 100 auto matmul_result = in SoftmaxCrossEntropyWithLogitsGrad() local
103 auto squeeze_result = Squeeze(scope, matmul_result, Squeeze::Axis(axis)); in SoftmaxCrossEntropyWithLogitsGrad()
|