Searched refs:PropagateTensorShapeOnNode (Results 1 – 1 of 1) sorted by relevance
729 if (PropagateTensorShapeOnNode(node, insert_expands)) { in propagateNode()768 bool PropagateTensorShapeOnNode(Node* node, bool insert_expands) { in PropagateTensorShapeOnNode() function in torch::jit::__anon9cb44b210111::ShapePropagator