Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_ops_to_mkldnn.cpp1146 RemoveTensorMutation(graph, [](Node* node_to_functionalize) { in ConvertFrozenOpsToMKLDNN() argument
1161 return mkldnn_ops.count(node_to_functionalize->kind()) != 0; in ConvertFrozenOpsToMKLDNN()