Searched defs:mat2_size (Results 1 – 5 of 5) sorted by relevance
673 const auto mat2_size = mat2_sizes_ref.toDimVector(); in computeFlops() local714 const auto mat2_size = mat2_sizes_ref.toDimVector(); in computeFlops() local
73 int confirm_mult_size(const std::vector<int>& mat1_size, const std::vector<int>& mat2_size) { in confirm_mult_size()
435 int32_t mat2_size = embedding_size * 4 * hunits; in LSTMData() local
2753 std::vector<int64_t> mat2_size = {K, N}; in test_mm() local