Home
last modified time | relevance | path

Searched refs:b_matrix (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_ops.py193 b_matrix = b._matrix if isinstance(b, SparseMatrix) else b
194 with ops.name_scope(name, "SparseMatrixMatMul", [a_matrix, b_matrix]):
201 b_matrix,
211 c_handle = matmul_shape_inference(a_matrix, b_matrix, c, transpose_a,
227 b_matrix,
234 b_matrix,
242 b_matrix,
/external/tensorflow/tensorflow/core/kernels/
Dgemm_functors.h102 typename tensorflow::TTypes<const T2>::Matrix b_matrix(b, k, n); in operator()
109 a_matrix.contract(b_matrix, dim_pair); in operator()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dadd_op.cc203 const CSRSparseMatrix* b_matrix; in Compute() local
205 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 1, &b_matrix)); in Compute()
213 ctx, b_matrix->dtype() == DataTypeToEnum<T>::value, in Compute()
215 DataTypeString(b_matrix->dtype()), " vs. ", in Compute()
236 OP_REQUIRES_OK(ctx, add_functor(*a_matrix, *b_matrix, &c_matrix)); in Compute()
Dsparse_mat_mul_op.cc316 const CSRSparseMatrix* b_matrix; in Compute() local
318 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 1, &b_matrix)); in Compute()
325 ctx, b_matrix->dtype() == DataTypeToEnum<T>::value, in Compute()
327 DataTypeString(b_matrix->dtype()), " vs. ", in Compute()
333 auto b_dense_shape = b_matrix->dense_shape().vec<int64>(); in Compute()
411 b_input_matrix = b_matrix; in Compute()
415 ctx, transpose(ctx, conjugate_b_, *b_matrix, &b_matrix_transposed)); in Compute()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1052 auto& b_matrix = op_info.inputs(1); in CountMatMulOperations() local
1072 MaybeGetMinimumShape(b_matrix.shape(), 2, found_unknown_shapes); in CountMatMulOperations()
1179 OpInfo::TensorProperties* b_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum() local
1180 b_matrix->set_dtype(b_input.dtype()); in GenerateBatchMatmulContextFromEinsum()
1181 TensorShapeProto* b_matrix_shape = b_matrix->mutable_shape(); in GenerateBatchMatmulContextFromEinsum()
1335 OpInfo::TensorProperties* b_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local
1336 b_matrix->set_dtype(b_input.dtype()); in CountBatchMatMulOperations()
1337 TensorShapeProto* b_matrix_shape = b_matrix->mutable_shape(); in CountBatchMatMulOperations()
1837 auto b_matrix = op_info.inputs(3); in PredictSparseTensorDenseMatMul() local
1839 MaybeGetMinimumShape(b_matrix.shape(), 2, &found_unknown_shapes); in PredictSparseTensorDenseMatMul()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2500 const DeviceMemory<T> &b_matrix = *b_ptrs_to_wrappers[b]; in DoBlasInternalImpl() local
2503 lda, b_matrix, ldb, beta, c_matrix, ldc); in DoBlasInternalImpl()
2635 const auto *b_matrix = in DoBlasInternalImpl() local
2643 a_matrix, SE_CUDA_DATA_HALF, lda, b_matrix, SE_CUDA_DATA_HALF, ldb, in DoBlasInternalImpl()