Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dsymbolic_shape_analysis.cpp644 std::vector<SSArgument> getNodeInputShapes(Node* n, const AliasDb& db) { in getNodeInputShapes() function
774 auto input_shapes = getNodeInputShapes(n, db); in PropagateShapesWithShapeFunction()