Searched defs:self_size0 (Results 1 – 2 of 2) sorted by relevance
311 const int64_t &self_size0 = self_shape[0], &self_size1 = self_shape[1], in bmm_nested_cuda() local356 const int64_t &self_size0 = self_shape[0]; in bmm_nested_cuda() local
45 const int64_t& self_size0 = self_shape[0], & self_size1 = self_shape[1], in bmm_nested() local