Searched defs:s_dim (Results 1 – 2 of 2) sorted by relevance
134 def _valid(in_dim, k_dim, s_dim): argument146 def _same(in_dim, s_dim): argument
815 int s_dim = 1; in CountBatchMatMulOperations() local