Searched defs:ShapeVisitor (Results 1 – 2 of 2) sorted by relevance
61 struct ShapeVisitor { struct62 ShapeVisitor(SymbolicExprsMap *symbolicExprsMap, in ShapeVisitor() argument67 void visit(ShapeOrValueInfo requestedInfo) { in visit()242 void backwardAssumingShape(Value op) { in backwardAssumingShape()251 void forwardAssumingShape(Value op) { in forwardAssumingShape()260 void backwardBroadcast(shape::BroadcastOp op) { in backwardBroadcast()265 void forwardBroadcast(shape::BroadcastOp op) { in forwardBroadcast()322 void backwardDynamicBroadcastInDimShape(mhlo::DynamicBroadcastInDimOp op) { in backwardDynamicBroadcastInDimShape()327 void forwardDynamicBroadcastInDimShape(mhlo::DynamicBroadcastInDimOp op) { in forwardDynamicBroadcastInDimShape()331 void backwardDynamicReshapeShape(mhlo::DynamicReshapeOp op) { in backwardDynamicReshapeShape()[all …]
64 using ShapeVisitor = variable