Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dshape_analysis.cpp729 if (PropagateTensorShapeOnNode(node, insert_expands)) { in propagateNode()
768 bool PropagateTensorShapeOnNode(Node* node, bool insert_expands) { in PropagateTensorShapeOnNode() function in torch::jit::__anon9cb44b210111::ShapePropagator