Searched refs:nonzero_numpy (Results 1 – 8 of 8) sorted by relevance
102 case aten::nonzero_numpy: in FuseWithListUnpack()
165 return self.nonzero_numpy(); in dispatch_nonzero_numpy()
390 nonzero_numpy, misc
577 return condition.nonzero_numpy(); in where()
2525 std::vector<Tensor> nonzero_numpy(const Tensor& self) { in nonzero_numpy() function
299 def nonzero_numpy(g: jit_utils.GraphContext, input, _outputs=None): function
4994 def nonzero_numpy(g: jit_utils.GraphContext, input, _outputs=None): function
455 return self.nonzero_numpy(); in dispatch_nonzero_numpy()