Searched defs:max_pool2d (Results 1 – 6 of 6) sorted by relevance
36 const torch::jit::Node* max_pool2d = find_node_by_kind(graph, "aten::max_pool2d"); in write() local
131 def max_pool2d(attrs, args): function
295 def max_pool2d(attrs, args): function
225 def max_pool2d(data, function
774 def max_pool2d( function