Searched refs:InferShapeForTensorListInitOps (Results 1 – 1 of 1) sorted by relevance
780 bool InferShapeForTensorListInitOps(Operation* op);1237 bool ShapeInference::InferShapeForTensorListInitOps(Operation* op) { in InferShapeForTensorListInitOps() function in mlir::TF::ShapeInference2108 if (IsTensorListInitOp(op) && InferShapeForTensorListInitOps(op)) return true; in InferShapeForSingleOperation()