| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mul_op.cc | 46 const CSRSparseMatrix* a_matrix; in Compute() local
47 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &a_matrix)); in Compute()
50 OP_REQUIRES(ctx, a_matrix->dtype() == b_t.dtype(), in Compute()
53 DataTypeString(a_matrix->dtype()), in Compute()
58 const Tensor& a_dense_shape_t = a_matrix->dense_shape(); in Compute()
64 ((a_matrix->dims() == 3) && (b_t.dim_size(0) == batch_size) && in Compute()
86 OP_REQUIRES_OK(ctx, csrmul_scalar.Compute(ctx, *a_matrix, b, &c_matrix)); in Compute()
92 csrmul_batch_vec.Compute(ctx, *a_matrix, b, &c_matrix)); in Compute()
|
| D | sparse_mat_mul_op.cc | 315 const CSRSparseMatrix* a_matrix; in Compute() local
317 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &a_matrix)); in Compute()
320 ctx, a_matrix->dtype() == DataTypeToEnum<T>::value, in Compute()
322 DataTypeString(a_matrix->dtype()), " vs. ", in Compute()
332 auto a_dense_shape = a_matrix->dense_shape().vec<int64>(); in Compute()
350 const int batch_size = a_matrix->batch_size(); in Compute()
396 a_input_matrix = a_matrix; in Compute()
400 ctx, transpose(ctx, conjugate_a_, *a_matrix, &a_matrix_transposed)); in Compute()
|
| D | mat_mul_op.cc | 500 const CSRSparseMatrix* a_matrix; in Compute() local
501 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &a_matrix)); in Compute()
507 this->ValidateInputs(*a_matrix, b_t, &rank, &batch_size)); in Compute()
509 const Tensor& a_dense_shape_t = a_matrix->dense_shape(); in Compute()
574 auto a_row_ptr = a_matrix->row_pointers_vec(i); in Compute()
575 auto a_col_ind = a_matrix->col_indices_vec(i); in Compute()
576 auto a_values = a_matrix->values_vec<T>(i); in Compute()
624 a_input_matrix = a_matrix; in Compute()
627 OP_REQUIRES_OK(ctx, transpose(ctx, this->conjugate_a_, *a_matrix, in Compute()
|
| D | add_op.cc | 202 const CSRSparseMatrix* a_matrix; in Compute() local
204 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &a_matrix)); in Compute()
208 ctx, a_matrix->dtype() == DataTypeToEnum<T>::value, in Compute()
210 DataTypeString(a_matrix->dtype()), " vs. ", in Compute()
236 OP_REQUIRES_OK(ctx, add_functor(*a_matrix, *b_matrix, &c_matrix)); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_ops.py | 192 a_matrix = a._matrix if isinstance(a, SparseMatrix) else a
194 with ops.name_scope(name, "SparseMatrixMatMul", [a_matrix, b_matrix]):
200 a_matrix,
211 c_handle = matmul_shape_inference(a_matrix, b_matrix, c, transpose_a,
217 a_matrix,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gemm_functors.h | 101 typename tensorflow::TTypes<const T1>::Matrix a_matrix(a, m, k); in operator()
109 a_matrix.contract(b_matrix, dim_pair); in operator()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1051 auto& a_matrix = op_info.inputs(0); in CountMatMulOperations() local
1070 MaybeGetMinimumShape(a_matrix.shape(), 2, found_unknown_shapes); in CountMatMulOperations()
1175 OpInfo::TensorProperties* a_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum() local
1176 TensorShapeProto* a_matrix_shape = a_matrix->mutable_shape(); in GenerateBatchMatmulContextFromEinsum()
1177 a_matrix->set_dtype(a_input.dtype()); in GenerateBatchMatmulContextFromEinsum()
1327 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local
1328 a_matrix->set_dtype(a_input.dtype()); in CountBatchMatMulOperations()
1329 TensorShapeProto* a_matrix_shape = a_matrix->mutable_shape(); in CountBatchMatMulOperations()
|
| /external/eigen/Eigen/src/Eigenvalues/ |
| D | SelfAdjointEigenSolver.h | 400 ::compute(const EigenBase<InputType>& a_matrix, int options)
404 const InputType &matrix(a_matrix.derived());
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 2499 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; in DoBlasInternalImpl() local
2502 bool ok = DoBlasGemm(stream, transa, transb, m, n, k, alpha, a_matrix, in DoBlasInternalImpl()
2633 const auto *a_matrix = in DoBlasInternalImpl() local
2643 a_matrix, SE_CUDA_DATA_HALF, lda, b_matrix, SE_CUDA_DATA_HALF, ldb, in DoBlasInternalImpl()
|