Searched refs:matmul_result (Results 1 – 2 of 2) sorted by relevance
125 Matrix matmul_result(m, n); in TEST() local126 matmul_result.setZero(); in TEST()127 matmul_result = lhs_mat * rhs_mat; in TEST()132 Scalar matmul = matmul_result(i % m, i / m); in TEST()
100 auto matmul_result = in SoftmaxCrossEntropyWithLogitsGrad() local103 auto squeeze_result = Squeeze(scope, matmul_result, Squeeze::Axis(axis)); in SoftmaxCrossEntropyWithLogitsGrad()