Searched defs:SymbolicShapeOpAnalyzer (Results 1 – 1 of 1) sorted by relevance
281 struct SymbolicShapeOpAnalyzer { struct282 std::shared_ptr<Graph> shape_compute_graph_;283 const FunctionSchema* schema_;284 std::vector<SSArgument> inputs_;290 void refineInputUnionTypes(const Node* parent_graph_node) { in refineInputUnionTypes()316 void substituteConstantInputs() { in substituteConstantInputs()390 void substituteSymbolicProperties( in substituteSymbolicProperties()516 void mergeSymbolicShapeSets(const std::vector<Value*>& symbolic_set) { in mergeSymbolicShapeSets()548 std::vector<c10::SymbolicShape> propagateShapesInGraph() { in propagateShapesInGraph()566 std::vector<c10::SymbolicShape> extractOutputShape( in extractOutputShape()[all …]