Home
last modified time | relevance | path

Searched defs:a_matrix (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dmul_op.cc46 const CSRSparseMatrix* a_matrix; in Compute() local
Dadd_op.cc202 const CSRSparseMatrix* a_matrix; in Compute() local
Dsparse_mat_mul_op.cc315 const CSRSparseMatrix* a_matrix; in Compute() local
Dmat_mul_op.cc500 const CSRSparseMatrix* a_matrix; in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1051 auto& a_matrix = op_info.inputs(0); in CountMatMulOperations() local
1175 OpInfo::TensorProperties* a_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum() local
1327 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2499 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; in DoBlasInternalImpl() local
2633 const auto *a_matrix = in DoBlasInternalImpl() local