Searched refs:shouldConsiderForMerge (Results 1 – 5 of 5) sorted by relevance
174 differentiable_nodes += shouldConsiderForMerge(curr); in buildWorkBlocks()243 bool shouldConsiderForMerge(Node* node) { in shouldConsiderForMerge() function in torch::jit::__anond27a20600111::SubgraphSlicer262 if (shouldConsiderForMerge(consumer)) { in scanNode()284 bool canMerge = shouldConsiderForMerge(producer) && in tryMerge()
1062 bool shouldConsiderForMerge(Node* node) { in shouldConsiderForMerge() function in torch::jit::__anonf4fb6ed70111::MKLDNNSubgraphSlicer1103 bool canMerge = shouldConsiderForMerge(consumer) && in tryMerge()
99 if (llgaHelper_.shouldConsiderForMerge(consumer)) { in scanNode()
55 bool shouldConsiderForMerge(Node* node);
463 if (!shouldConsiderForMerge(toMerge)) { in shouldMerge()499 bool LlgaGraphHelper::shouldConsiderForMerge(Node* node) { in shouldConsiderForMerge() function in torch::jit::fuser::onednn::LlgaGraphHelper