Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dimplementation_selector.cc131 void UpdateForwardIdentityNodeDtype(utils::MutableNodeView* forward_node, in UpdateForwardIdentityNodeDtype()
/aosp_15_r20/external/pytorch/torch/distributed/pipelining/
H A D_IR.py160 def assign_or_accumulate_grad(forward_node, grad_value): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dfunction_extraction.cpp1086 Node* NodeOfMostRecentScope(Node* forward_node) { in NodeOfMostRecentScope()