Searched defs:mat1_size (Results 1 – 5 of 5) sorted by relevance
672 const auto mat1_size = mat1_sizes_ref.toDimVector(); in computeFlops() local713 const auto mat1_size = mat1_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()
434 int32_t mat1_size = (num_index + 1) * embedding_size; in LSTMData() local
2752 std::vector<int64_t> mat1_size = {M, K}; in test_mm() local3086 std::vector<int64_t> mat1_size = {M, K}; in test_transpose_view_mm() local