Home
last modified time | relevance | path

Searched defs:fusion_node (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/cuda/
H A Dinterface.cpp79 void compileFusionGroup(Node* fusion_node) { in compileFusionGroup()
88 void runFusionGroup(const Node* fusion_node, Stack& stack) { in runFusionGroup()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dfusion.cpp209 Node* fusion_node, in createFusionGroup()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmulti_output_fusion.cc206 FusionCandidate& fusion_node = candidates_[get_candidate_id(fusion)]; in GetNewFusibles() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dtensorexpr_fuser.cpp598 Node* fusion_node) { in createFusionGroup()