Searched defs:rowwise (Results 1 – 3 of 3) sorted by relevance
68 self, local_module, dist_module, inp_size, rank0_only=True, rowwise=False argument
312 def colorize(self, rowwise=False): argument
550 EIGEN_DEVICE_FUNC inline ConstRowwiseReturnType rowwise() const { in rowwise() function