Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dmul_op.cc46 const CSRSparseMatrix* a_matrix; in Compute() local
H A Dadd_op.cc203 const CSRSparseMatrix* a_matrix; in Compute() local
H A Dsparse_mat_mul_op.cc315 const CSRSparseMatrix* a_matrix; in Compute() local
H A Dmat_mul_op.cc509 const CSRSparseMatrix* a_matrix; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas.cc1065 const auto *a_matrix = reinterpret_cast<const __nv_bfloat16 *>( in DoBlasInternalImpl() local
1312 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; in DoBlasInternalImpl() local
1442 const auto *a_matrix = reinterpret_cast<const __nv_bfloat16 *>( in DoBlasInternalImpl() local
1478 const auto *a_matrix = reinterpret_cast<const __half *>( in DoBlasInternalImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc1063 auto& a_matrix = op_info.inputs(0); in CountMatMulOperations() local
1187 OpInfo::TensorProperties* a_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum() local
1339 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local