Home
last modified time | relevance | path

Searched refs:wrap_node (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/torch/fx/experimental/
Dsym_node.py1027 def wrap_node(x): function
1068 return to_node(self, alternate_impl(wrap_node(self), wrap_node(other)))
1072 self, handle_sym_dispatch(op, (wrap_node(self), wrap_node(other)), {})
1140 return to_node(self, handle_sym_dispatch(op, (wrap_node(self),), {}))
1187 wrap_node(pred_node),
1188 wrap_node(then_node),
1189 wrap_node(else_node),
1228 self, handle_sym_dispatch(op, (wrap_node(self), ndigits), {})
1282 ([wrap_node(s) for s in sizes], [wrap_node(s) for s in strides]),
1332 return wrap_node(
[all …]
/external/pytorch/c10/core/
DSymbolicShapeMeta.cpp67 size_nodes.emplace_back(s.wrap_node(base)); in normalize_sym_sizes_strides()
70 stride_nodes.emplace_back(s.wrap_node(base)); in normalize_sym_sizes_strides()
DSymFloat.h39 SymNode wrap_node(const SymNode& base) const;
DSymBool.h34 SymNode wrap_node(const SymNode& base) const;
DSymBool.cpp11 SymNode SymBool::wrap_node(const SymNode& base) const { in wrap_node() function in c10::SymBool
DSymFloat.cpp14 SymNode SymFloat::wrap_node(const SymNode& base) const { in wrap_node() function in c10::SymFloat
DSymInt.cpp104 SymNode SymInt::wrap_node(const SymNode& base) const { in wrap_node() function in c10::SymInt
DSymInt.h121 SymNode wrap_node(const SymNode& base) const;
/external/pytorch/test/dynamo/
Dtest_activation_checkpointing.py500 wrap_node = find_first_node(cnt.graphs[0], tag_activation_checkpoint)
502 self.assertEqual(len(wrap_node.args), 4)
504 body_function = getattr(cnt.graphs[0], wrap_node.args[0].name)
534 wrap_node = find_first_node(cnt.graphs[0], tag_activation_checkpoint)
535 self.assertEqual(len(wrap_node.args), 3)
Dtest_higher_order_ops.py187 wrap_node = find_first_node(graph, wrap)
188 self.assertEqual(len(wrap_node.args), expected_num_wrap_args)
475 wrap_node = find_first_node(backend.graphs[0], wrap)
476 self.assertTrue(len(wrap_node.args), 3)
478 body_function = getattr(backend.graphs[0], wrap_node.args[0].name)
631 wrap_node = find_first_node(gm, wrap)
632 self.assertTrue(len(wrap_node.args), 3)
635 body_function = getattr(gm, wrap_node.args[0].name)
1980 wrap_node = find_first_node(backend.graphs[0], wrap)
1982 self.assertTrue(len(wrap_node.args), 3)
[all …]