Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dtensorexpr_fuser.cpp571 Node* getOrCreateTensorExprSubgraph(Node* n) { in getOrCreateTensorExprSubgraph() function in torch::jit::TensorExprFuser
603 fusion_node = getOrCreateTensorExprSubgraph(fusion_node); in createFusionGroup()
803 fusion_group = getOrCreateTensorExprSubgraph(fusion_group); in tryMerge()