Home
last modified time | relevance | path

Searched refs:matrix_b (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dbatch_matmul_op_test.py236 matrix_b = variables.Variable(
243 math_ops.matmul(matrix_a, matrix_b),
249 matrix_a.shape[:-2], matrix_b.shape[:-2])
253 matrix_b.shape[-2:])
258 array_ops.broadcast_to(matrix_b, broadcasted_b_shape)),
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc172 const Tensor& matrix_b = ctx->input(1); in Compute() local
176 OP_REQUIRES_OK(ctx, this->ValidateInputs(*sparse_matrix_a, matrix_b, &rank, in Compute()
182 int64 num_rhs_rows = matrix_b.dim_size(rank - 2); in Compute()
183 int64 num_rhs_cols = matrix_b.dim_size(rank - 1); in Compute()
190 const Tensor* rhs = &matrix_b; in Compute()
194 ctx, TransposeAndConjugateTensor(ctx, matrix_b, this->conjugate_b_, in Compute()
/external/deqp/external/openglcts/modules/gl/
Dgl4cGPUShaderFP64Tests.cpp11158 tcu::Matrix2d matrix_b(&matrix_b_data[0]); in getMatrixMultiplicationResult() local
11162 matrix_b = transpose(matrix_b); in getMatrixMultiplicationResult()
11163 result = matrix_a * matrix_b; in getMatrixMultiplicationResult()
11175 tcu::Matrix<double, 3, 2> matrix_b(&matrix_b_data[0]); in getMatrixMultiplicationResult() local
11180 matrix_b_transposed = transpose(matrix_b); in getMatrixMultiplicationResult()
11193 tcu::Matrix<double, 4, 2> matrix_b(&matrix_b_data[0]); in getMatrixMultiplicationResult() local
11198 matrix_b_transposed = transpose(matrix_b); in getMatrixMultiplicationResult()
11210 tcu::Matrix<double, 3, 3> matrix_b(&matrix_b_data[0]); in getMatrixMultiplicationResult() local
11214 matrix_b = transpose(matrix_b); in getMatrixMultiplicationResult()
11215 result = matrix_a * matrix_b; in getMatrixMultiplicationResult()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h2502 ConstMatrixRef matrix_b(b, n, k); in Conv()
2508 matrix_c.col(0).noalias() = matrix_a * matrix_b.row(0).transpose(); in Conv()
2511 matrix_c.row(0).noalias() = matrix_a.row(0) * matrix_b.transpose(); in Conv()
2514 matrix_c.noalias() = matrix_a * matrix_b.transpose(); in Conv()