Searched defs:functional_op (Results 1 – 4 of 4) sorted by relevance
93 def construct_functionalization_kernel(mutable_op, functional_op): argument
9 torch::Tensor functional_op(torch::Tensor& x) { in functional_op() function
12375 def functional_op(x): function12420 def functional_op(x): function12542 def functional_op(x): function
595 def _check_is_functional_variant(self, mutable_op, functional_op, args): argument