Searched defs:count_node (Results 1 – 5 of 5) sorted by relevance
68 NodeDef *count_node = graph_utils::AddScalarConstNode<int64_t>(-1, &graph); in TEST() local134 NodeDef *count_node = graph_utils::AddScalarConstNode<int64_t>(-1, &graph); in TEST() local204 NodeDef *count_node = graph_utils::AddScalarConstNode<int64_t>(-1, &graph); in TEST() local260 NodeDef *count_node = graph_utils::AddScalarConstNode<int64_t>(-1, &graph); in TEST() local
40 const auto& count_node = *graph.GetNode(take_node.input(1)); in IsTakeAll() local
94 def count_node(graph_module: torch.fx.GraphModule, target: torch.fx.node.Target) -> int: function
914 Node* count_node = nullptr; in AsGraphDefInternal() local
124 def count_node(self, graph_module: torch.fx.GraphModule, opname: str) -> int: member in TestEmit