Searched defs:a_matrix (Results 1 – 2 of 2) sorted by relevance
698 auto& a_matrix = op_info.inputs(0); in CountMatMulOperations() local841 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local
2258 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; in DoBlasGemmBatchedInternal() local2395 const auto *a_matrix = in DoBlasGemmStridedBatched() local