Home
last modified time | relevance | path

Searched defs:shouldConsiderForMerge (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/
Dcreate_autodiff_subgraphs.cpp243 bool shouldConsiderForMerge(Node* node) { in shouldConsiderForMerge() function in torch::jit::__anond27a20600111::SubgraphSlicer
Dfrozen_ops_to_mkldnn.cpp1062 bool shouldConsiderForMerge(Node* node) { in shouldConsiderForMerge() function in torch::jit::__anonf4fb6ed70111::MKLDNNSubgraphSlicer
/external/pytorch/torch/csrc/jit/codegen/onednn/
Dgraph_helper.cpp499 bool LlgaGraphHelper::shouldConsiderForMerge(Node* node) { in shouldConsiderForMerge() function in torch::jit::fuser::onednn::LlgaGraphHelper