Searched refs:matrix_b (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | batch_matmul_op_test.py | 236 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/ |
D | mat_mul_op.cc | 172 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/ |
D | gl4cGPUShaderFP64Tests.cpp | 11158 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/ |
D | legacy_optimized_ops.h | 2502 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()
|